Removing OpenMP from unit tests (#163)

[ROCm/rccl commit: 000bce6f27]
Этот коммит содержится в:
gilbertlee-amd
2019-12-20 11:41:56 -07:00
коммит произвёл GitHub
родитель a461b6d139
Коммит 71635198b8
11 изменённых файлов: 14 добавлений и 28 удалений
+2 -3
Просмотреть файл
@@ -5,7 +5,6 @@
************************************************************************/
#include "test_ReduceScatter.hpp"
#include <omp.h>
namespace CorrectnessTests
{
@@ -24,7 +23,7 @@ namespace CorrectnessTests
size_t const recvCount = dataset.numElements / dataset.numDevices;
// Launch the reduction (1 thread per GPU)
#pragma omp parallel for num_threads(numDevices)
ncclGroupStart();
for (int i = 0; i < numDevices; i++)
{
ncclReduceScatter(dataset.inputs[i],
@@ -32,7 +31,7 @@ namespace CorrectnessTests
recvCount, dataType, op,
comms[i], streams[i]);
}
ncclGroupEnd();
// Wait for reduction to complete
Synchronize();