From c8d3543d3f854f3da2672eae2b28bc33c8be15a3 Mon Sep 17 00:00:00 2001
From: Wenkai Du <43822138+wenkaidu@users.noreply.github.com>
Date: Tue, 15 Oct 2024 08:12:26 -0700
Subject: [PATCH] Add back missing net flush (#1376)

---
 src/graph/paths.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/graph/paths.cc b/src/graph/paths.cc
index a1f8d96896..de78d804d8 100644
--- a/src/graph/paths.cc
+++ b/src/graph/paths.cc
@@ -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();