[ROCm/rccl-tests commit: dc1ad4853d]
Этот коммит содержится в:
Wenkai Du
2021-09-22 08:43:01 -07:00
родитель 669a778efb
Коммит cc55343b88
+1 -1
Просмотреть файл
@@ -527,7 +527,7 @@ testResult_t startColl(struct threadArgs* args, ncclDataType_t type, ncclRedOp_t
size_t count = args->nbytes / wordSize(type);
// Try to change offset for each iteration so that we avoid cache effects and catch race conditions in ptrExchange
size_t totalnbytes = max(args->sendBytes, args->expectedBytes);
size_t totalnbytes = std::max(args->sendBytes, args->expectedBytes);
size_t steps = totalnbytes ? args->maxbytes / totalnbytes : 1;
size_t shift = totalnbytes * (iter % steps);