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

[ROCm/rccl-tests commit: 5ee7a08994]
This commit is contained in:
Wenkai Du
2023-10-16 12:13:50 -07:00
committed by GitHub
parent 2e12f0cfce
commit ccad358bc9
+1 -1
View File
@@ -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));