Warm up both out-of-place and in-place collectives (#51)

Este cometimento está contido em:
Wenkai Du criou e GitHub cometeu 2023-10-16 12:13:50 -07:00
1 ascendente a6593375bc
cometimento 5ee7a08994
1 ficheiro modificado
+1 -1
+1 -1
Ver ficheiro
@@ -641,7 +641,7 @@ testResult_t TimeTest(struct threadArgs* args, ncclDataType_t type, const char*
// Warm-up for small size // Warm-up for small size
setupArgs(args->minbytes, type, args); setupArgs(args->minbytes, type, args);
for (int iter = 0; iter < warmup_iters; iter++) { for (int iter = 0; iter < warmup_iters; iter++) {
TESTCHECK(startColl(args, type, op, root, 0, iter)); TESTCHECK(startColl(args, type, op, root, iter < warmup_iters/2 ? 0 : 1, iter));
} }
TESTCHECK(completeColl(args)); TESTCHECK(completeColl(args));