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

Этот коммит содержится в:
Wenkai Du
2023-10-16 12:13:50 -07:00
коммит произвёл GitHub
родитель a6593375bc
Коммит 5ee7a08994
+1 -1
Просмотреть файл
@@ -641,7 +641,7 @@ testResult_t TimeTest(struct threadArgs* args, ncclDataType_t type, const char*
// Warm-up for small size
setupArgs(args->minbytes, type, args);
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));