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

[ROCm/rccl commit: f6f5e16fe6]
Tento commit je obsažen v:
Stanley Tsang
2021-06-10 14:35:39 -06:00
odevzdal GitHub
rodič a24e180296
revize dd98f1762a
11 změnil soubory, kde provedl 28 přidání a 13 odebrání
+1 -1
Zobrazit soubor
@@ -14,7 +14,7 @@ namespace CorrectnessTests
// Allocate data
Dataset dataset;
dataset.Initialize(numDevices, numElements, dataType, inPlace);
dataset.Initialize(numDevices, numElements, dataType, inPlace, ncclCollReduce);
// Test each possible root
for (int root = 0; root < numDevices; root++)