Add back missing net flush (#1376)
This commit is contained in:
@@ -458,6 +458,7 @@ ncclResult_t ncclTopoNeedFlush(struct ncclTopoSystem* system, int64_t busId, int
|
|||||||
NCCLCHECK(ncclTopoIdToIndex(system, GPU, busId, &g));
|
NCCLCHECK(ncclTopoIdToIndex(system, GPU, busId, &g));
|
||||||
struct ncclTopoNode* gpu = system->nodes[GPU].nodes+g;
|
struct ncclTopoNode* gpu = system->nodes[GPU].nodes+g;
|
||||||
#if defined(__HIP_PLATFORM_AMD__) || defined(__HIPCC__)
|
#if defined(__HIP_PLATFORM_AMD__) || defined(__HIPCC__)
|
||||||
|
*flush = 1;
|
||||||
#else
|
#else
|
||||||
// Flush is required on Ampere and earlier
|
// Flush is required on Ampere and earlier
|
||||||
*flush = gpu->gpu.cudaCompCap < 90 ? 1 : ncclParamNetForceFlush();
|
*flush = gpu->gpu.cudaCompCap < 90 ? 1 : ncclParamNetForceFlush();
|
||||||
|
|||||||
Reference in New Issue
Block a user