Fixing bug with ExtractSubDataset function not fully initializing subdataset (#390)
This commit is contained in:
1 parent
eea7b24058
commit
f6f5e16fe6
11 files changed
+28
-13
No files matched your search
@@ -20,9 +20,16 @@ namespace CorrectnessTests
|
||||
|
||||
// Create multiple datasets for combined 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]);
|
||||
}
|
||||
|
||||
|
||||
Reference in new issue
Block a user