Add back missing net flush (#1376)

Este commit está contenido en:
Wenkai Du authored and GitHub committed 2024-10-15 08:12:26 -07:00
1 padre 62d10fdc25
commit c8d3543d3f
1 file changed
+1
+1
Ver fichero
@@ -458,6 +458,7 @@ ncclResult_t ncclTopoNeedFlush(struct ncclTopoSystem* system, int64_t busId, int
NCCLCHECK(ncclTopoIdToIndex(system, GPU, busId, &g));
struct ncclTopoNode* gpu = system->nodes[GPU].nodes+g;
#if defined(__HIP_PLATFORM_AMD__) || defined(__HIPCC__)
*flush = 1;
#else
// Flush is required on Ampere and earlier
*flush = gpu->gpu.cudaCompCap < 90 ? 1 : ncclParamNetForceFlush();