diff --git a/src/all_reduce.cu b/src/all_reduce.cu index 5302f86..a3d33f9 100644 --- a/src/all_reduce.cu +++ b/src/all_reduce.cu @@ -65,8 +65,10 @@ 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 defined(RCCL_FLOAT8) if((type == ncclFp8E4M3 || type == ncclFp8E5M2) && op == ncclProd) return testSuccess; +#endif if ((int)type != -1) { type_count = 1; @@ -90,8 +92,10 @@ testResult_t AllReduceRunTest(struct threadArgs* args, int root, ncclDataType_t for (int i=0; i