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

UnitTests: made reduceScatter run a smaller amount of tests (#691)

Browse Source 
[ROCm/rccl commit: d119c0886e]
This commit is contained in:
akolliasAMD
2023-02-21 16:21:24 -07:00
committed by GitHub
parent 1f42d485f8
commit 3b6b34705a
3 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rccl/test/ReduceScatter_InPlace.cpp
+3 -3
View File
@@ -13,10 +13,10 @@ namespace RcclUnitTesting
// Configuration
std::vector<ncclFunc_t> const funcTypes = {ncclCollReduceScatter};
std::vector<ncclDataType_t> const dataTypes = {ncclInt8, ncclInt32, ncclInt64, ncclFloat16};
std::vector<ncclRedOp_t> const redOps = {ncclSum, ncclProd};
std::vector<ncclDataType_t> const dataTypes = {ncclInt8, ncclInt32, ncclInt64};
std::vector<ncclRedOp_t> const redOps = {ncclProd};
std::vector<int> const roots = {0};
std::vector<int> const numElements = {1048576, 53327, 5461, 1024};
std::vector<int> const numElements = {1048576, 53327, 1024};;
std::vector<bool> const inPlaceList = {true};
std::vector<bool> const managedMemList = {false};
std::vector<bool> const useHipGraphList = {false, true};
projects/rccl/test/ReduceScatter_ManagedMem.cpp
+1 -1
View File
@@ -14,7 +14,7 @@ namespace RcclUnitTesting
// Configuration
std::vector<ncclFunc_t> const funcTypes = {ncclCollReduceScatter};
std::vector<ncclDataType_t> const dataTypes = {ncclUint8, ncclUint32, ncclUint64};
std::vector<ncclRedOp_t> const redOps = {ncclSum};
std::vector<ncclRedOp_t> const redOps = {ncclAvg};
std::vector<int> const roots = {0};
std::vector<int> const numElements = {1048576, 53327, 1024};
std::vector<bool> const inPlaceList = {false};
projects/rccl/test/ReduceScatter_OutOfPlace.cpp
+3 -3
View File
@@ -13,10 +13,10 @@ namespace RcclUnitTesting
// Configuration
std::vector<ncclFunc_t> const funcTypes = {ncclCollReduceScatter};
std::vector<ncclDataType_t> const dataTypes = {ncclFloat16, ncclFloat32, ncclFloat64, ncclBfloat16};
std::vector<ncclRedOp_t> const redOps = {ncclMin, ncclMax, ncclAvg};
std::vector<ncclDataType_t> const dataTypes = {ncclFloat32, ncclFloat64, ncclBfloat16};
std::vector<ncclRedOp_t> const redOps = {ncclMax};
std::vector<int> const roots = {0};
std::vector<int> const numElements = {1048576, 53327, 5461, 1024};
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};