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

이 커밋은 다음에 포함됨:
akolliasAMD
2023-02-21 16:21:24 -07:00
커밋한 사람 GitHub
부모 86e7b71234
커밋 d119c0886e
3개의 변경된 파일7개의 추가작업 그리고 7개의 파일을 삭제
+3 -3
파일 보기
@@ -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};