removing FP8 product from allReduce test cases (#97)
* removing FP8 product from allReduce test cases --------- Co-authored-by: Marzieh Berenjkoub <mberenjk@amd.com>
Этот коммит содержится в:
@@ -65,6 +65,8 @@ testResult_t AllReduceRunTest(struct threadArgs* args, int root, ncclDataType_t
|
||||
ncclRedOp_t *run_ops;
|
||||
const char **run_typenames, **run_opnames;
|
||||
int type_count, op_count;
|
||||
if((type == ncclFp8E4M3 || type == ncclFp8E5M2) && op == ncclProd)
|
||||
return testSuccess;
|
||||
|
||||
if ((int)type != -1) {
|
||||
type_count = 1;
|
||||
@@ -88,6 +90,8 @@ testResult_t AllReduceRunTest(struct threadArgs* args, int root, ncclDataType_t
|
||||
|
||||
for (int i=0; i<type_count; i++) {
|
||||
for (int j=0; j<op_count; j++) {
|
||||
if((i == ncclFp8E4M3 || i == ncclFp8E5M2) && j == ncclProd)
|
||||
continue;
|
||||
TESTCHECK(TimeTest(args, run_types[i], run_typenames[i], run_ops[j], run_opnames[j], -1));
|
||||
}
|
||||
}
|
||||
|
||||
Ссылка в новой задаче
Block a user