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
파일 보기
@@ -11,8 +11,6 @@
#include "test_Reduce.hpp"
#include "test_ReduceScatter.hpp"
#include <omp.h>
namespace CorrectnessTests
{
TEST_P(CombinedCallsCorrectnessTest, Correctness)
@@ -38,7 +36,7 @@ namespace CorrectnessTests
size_t const byteCount = datasets[0].NumBytes() / numDevices;
size_t const elemCount = numElements / numDevices;
#pragma omp parallel for num_threads(numDevices)
ncclGroupStart();
for (int i = 0; i < numDevices; i++)
{
ncclAllGather((int8_t *)datasets[0].inputs[i] + (i * byteCount),
@@ -63,6 +61,7 @@ namespace CorrectnessTests
elemCount, dataType, op,
comms[i], streams[i]);
}
ncclGroupEnd();
// Wait for reduction to complete
Synchronize();