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

[ROCm/rccl commit: f6f5e16fe6]
This commit is contained in:
Stanley Tsang
2021-06-10 14:35:39 -06:00
کامیت شده توسط GitHub
والد a24e180296
کامیت dd98f1762a
11فایلهای تغییر یافته به همراه28 افزوده شده و 13 حذف شده
@@ -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]);
}