Fixing bug with ExtractSubDataset function not fully initializing subdataset (#390)

[ROCm/rccl commit: f6f5e16fe6]
Цей коміт міститься в:
Stanley Tsang
2021-06-10 14:35:39 -06:00
зафіксовано GitHub
джерело a24e180296
коміт dd98f1762a
11 змінених файлів з 28 додано та 13 видалено
+3 -3
Переглянути файл
@@ -15,9 +15,9 @@ namespace CorrectnessTests
// Prepare input / output / expected results
Dataset dataset1, dataset2, dataset3;
dataset1.Initialize(numDevices, numElements, dataType, inPlace);
dataset2.Initialize(numDevices, numElements, dataType, inPlace);
dataset3.Initialize(numDevices, numElements, dataType, inPlace);
dataset1.Initialize(numDevices, numElements, dataType, inPlace, ncclCollAllReduce);
dataset2.Initialize(numDevices, numElements, dataType, inPlace, ncclCollAllReduce);
dataset3.Initialize(numDevices, numElements, dataType, inPlace, ncclCollAllReduce);
FillDatasetWithPattern(dataset1);
FillDatasetWithPattern(dataset2);
FillDatasetWithPattern(dataset3);