Add back missing net flush (#1376)

Cette révision appartient à :
Wenkai Du a créé et GitHub a révisé 2024-10-15 08:12:26 -07:00
1 Parent 62d10fdc25
révision c8d3543d3f
1 fichier modifié
+1
+1
Voir le fichier
@@ -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();