Add support for ncclAvg operation

This commit is contained in:
David Addison
2021-06-17 14:08:43 -07:00
szülő e37545e491
commit cde7e769c1
8 fájl változott, egészen pontosan 90 új sor hozzáadva és 25 régi sor törölve
+2 -2
Fájl megtekintése
@@ -83,7 +83,7 @@ testResult_t AllReduceRunTest(struct threadArgs* args, int root, ncclDataType_t
run_types = &type;
run_typenames = &typeName;
} else {
type_count = ncclNumTypes;
type_count = test_typenum;
run_types = test_types;
run_typenames = test_typenames;
}
@@ -93,7 +93,7 @@ testResult_t AllReduceRunTest(struct threadArgs* args, int root, ncclDataType_t
run_ops = &op;
run_opnames = &opName;
} else {
op_count = ncclNumOps;
op_count = test_opnum;
run_ops = test_ops;
run_opnames = test_opnames;
}