Skip to content

Commit 77ae744

Browse files
mberenjkMarzieh Berenjkoub
andauthored
removing FP8 product from allReduce test cases (ROCm#97)
* removing FP8 product from allReduce test cases --------- Co-authored-by: Marzieh Berenjkoub <mberenjk@amd.com>
1 parent ae3e635 commit 77ae744

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

‎src/all_reduce.cu‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -65,6 +65,8 @@ testResult_t AllReduceRunTest(struct threadArgs* args, int root, ncclDataType_t
6565
ncclRedOp_t *run_ops;
6666
const char **run_typenames, **run_opnames;
6767
int type_count, op_count;
68+
if((type == ncclFp8E4M3 || type == ncclFp8E5M2) && op == ncclProd)
69+
return testSuccess;
6870

6971
if ((int)type != -1) {
7072
type_count = 1;
@@ -88,6 +90,8 @@ testResult_t AllReduceRunTest(struct threadArgs* args, int root, ncclDataType_t
8890

8991
for (int i=0; i<type_count; i++) {
9092
for (int j=0; j<op_count; j++) {
93+
if((i == ncclFp8E4M3 || i == ncclFp8E5M2) && j == ncclProd)
94+
continue;
9195
TESTCHECK(TimeTest(args, run_types[i], run_typenames[i], run_ops[j], run_opnames[j], -1));
9296
}
9397
}

0 commit comments

Comments
 (0)