Fix the sync bug in stream ref wait

NVIDIA · Dec 21, 2023 · 44f6821 · 44f6821
1 parent 2165845
commit 44f6821
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/libcudacxx/include/cuda/stream_ref b/libcudacxx/include/cuda/stream_ref
@@ -139,7 +139,7 @@ public:
    */
   void wait() const
   {
-    const auto __result = ::cudaStreamQuery(get());
+    const auto __result = ::cudaStreamSynchronize(get());
     switch (__result)
     {
       case ::cudaSuccess: