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

Graph unit tests (#656)

Browse Source 
* Adding hipGraph unit tests

[ROCm/rccl commit: faed69f9fc]
This commit is contained in:
gilbertlee-amd authored and GitHub committed 2022-12-01 10:28:42 -07:00
1 parent 2f1882c75e
commit 5871811d34
31 files changed
+399 -251
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

projects/rccl/test/Gather_OutOfPlace.cpp
+10 -8
View File
@@ -12,15 +12,17 @@ namespace RcclUnitTesting
TestBed testBed;
// Configuration
std::vector<ncclFunc_t> const funcTypes = {ncclCollGather};
std::vector<ncclDataType_t> const dataTypes = {ncclFloat32, ncclFloat64, ncclBfloat16};
std::vector<ncclRedOp_t> const redOps = {ncclSum};
std::vector<int> const roots = {1};
std::vector<int> const numElements = {1048576, 53327, 1024};
std::vector<bool> const inPlaceList = {false};
std::vector<bool> const managedMemList = {false};
std::vector<ncclFunc_t> const funcTypes = {ncclCollGather};
std::vector<ncclDataType_t> const dataTypes = {ncclFloat32, ncclFloat64, ncclBfloat16};
std::vector<ncclRedOp_t> const redOps = {ncclSum};
std::vector<int> const roots = {1};
std::vector<int> const numElements = {1048576, 53327, 1024};
std::vector<bool> const inPlaceList = {false};
std::vector<bool> const managedMemList = {false};
std::vector<bool> const useHipGraphList = {false, true};
testBed.RunSimpleSweep(funcTypes, dataTypes, redOps, roots, numElements, inPlaceList, managedMemList);
testBed.RunSimpleSweep(funcTypes, dataTypes, redOps, roots, numElements,
inPlaceList, managedMemList, useHipGraphList);
testBed.Finalize();
}
}