Adding FP16 cases to unit tests(#1093)
Signed-off-by: Tim Hu <timhu102@amd.com>
[ROCm/rccl commit: 0d06b0f1de]
이 커밋은 다음에 포함됨:
1 부모
874998033f
커밋
826d20495f
6 파일이 변경됨
+63
-14
검색과 일치하는 파일 없음
@@ -16,7 +16,7 @@ namespace RcclUnitTesting
|
||||
|
||||
// Configuration
|
||||
std::vector<ncclFunc_t> const funcTypes = {ncclCollAllToAll};
|
||||
std::vector<ncclDataType_t> const dataTypes = {ncclFloat32};
|
||||
std::vector<ncclDataType_t> const dataTypes = {ncclFloat16, ncclFloat32};
|
||||
std::vector<ncclRedOp_t> const redOps = {ncclSum};
|
||||
std::vector<int> const roots = {0};
|
||||
std::vector<int> const numElements = {1048576, 1024};
|
||||
|
||||
새 이슈에서 참조
사용자 차단