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

2021-06-10 14:35:39 -06:00
parent eea7b24058
commit f6f5e16fe6
11 changed files with 28 additions and 13 deletions
@@ -30,7 +30,7 @@ namespace CorrectnessTests

        // Prepare input / output / expected results
        Dataset dataset;
-        dataset.Initialize(numDevices, numElements, dataType, inPlace);
+        dataset.Initialize(numDevices, numElements, dataType, inPlace, ncclCollBroadcast);
        FillDatasetWithPattern(dataset);

        int root = 0;