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
@@ -19,9 +19,16 @@ namespace CorrectnessTests

        // Create multiple datasets for group operation
        std::vector<Dataset> datasets(5);
+        std::vector<ncclFunc_t> ncclFuncs(5);
+        ncclFuncs.push_back(ncclCollAllGather);
+        ncclFuncs.push_back(ncclCollAllReduce);
+        ncclFuncs.push_back(ncclCollBroadcast);
+        ncclFuncs.push_back(ncclCollReduce);
+        ncclFuncs.push_back(ncclCollReduceScatter);
+
        for (int i = 0; i < datasets.size(); i++)
        {
-            datasets[i].Initialize(numDevices, numElements, dataType, inPlace);
+            datasets[i].Initialize(numDevices, numElements, dataType, inPlace, ncclFuncs[i]);
            FillDatasetWithPattern(datasets[i]);
        }