From 2288e9ae80615b69cbb6e63d7b40575205c6f6eb Mon Sep 17 00:00:00 2001
From: Wenkai Du <43822138+wenkaidu@users.noreply.github.com>
Date: Thu, 15 Dec 2022 10:16:46 -0800
Subject: [PATCH] Switch to hipLaunchHostFunc for HIP graph (#667)

---
 src/misc/strongstream.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/misc/strongstream.cc b/src/misc/strongstream.cc
index 970cb34f81..23bbbd9c13 100644
--- a/src/misc/strongstream.cc
+++ b/src/misc/strongstream.cc
@@ -258,7 +258,7 @@ ncclResult_t ncclStrongStreamLaunchHost(
     }
     ss->serialEventNeedsRecord = true;
   #else
-    CUDACHECK(hipStreamAddCallback(ss->cudaStream, (hipStreamCallback_t)fn, arg, 0));
+    CUDACHECK(cudaLaunchHostFunc(ss->cudaStream, fn, arg));
   #endif
   return ncclSuccess;
 }