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));