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

[ROCm/rccl commit: f6f5e16fe6]
2021-06-10 14:35:39 -06:00
parent a24e180296
commit dd98f1762a
11 changed files with 28 additions and 13 deletions
@@ -14,7 +14,7 @@ namespace CorrectnessTests

        // Allocate data
        Dataset dataset;
-        dataset.Initialize(numDevices, numElements, dataType, inPlace);
+        dataset.Initialize(numDevices, numElements, dataType, inPlace, ncclCollBroadcast);

        // Test each possible root
        for (int root = 0; root < numDevices; root++)