AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Stanley Tsang
2021-06-10 14:35:39 -06:00
committed by GitHub
parent eea7b24058
commit f6f5e16fe6
11 changed files with 28 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/test_AllGather.cpp
+2 -2
View File
@@ -14,7 +14,7 @@ namespace CorrectnessTests
// Prepare input / output / expected results
Dataset dataset;
dataset.Initialize(numDevices, numElements, dataType, inPlace);
dataset.Initialize(numDevices, numElements, dataType, inPlace, ncclCollAllGather);
FillDatasetWithPattern(dataset);
ComputeExpectedResults(dataset);
@@ -46,7 +46,7 @@ namespace CorrectnessTests
// Allocate dataset
Dataset dataset;
dataset.Initialize(numDevices, numElements, dataType, inPlace);
dataset.Initialize(numDevices, numElements, dataType, inPlace, ncclCollAllGather);
// Loop over several offsets (so that device pointers are not aligned)
for (int firstElement = 1; firstElement <= 11; firstElement += 2)