Use DebugSyncStream where appropriate.

NVIDIA · May 18, 2022 · 7485284 · 7485284
1 parent 0f9587a
commit 7485284
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/thrust/system/cuda/detail/core/agent_launcher.h b/thrust/system/cuda/detail/core/agent_launcher.h
@@ -514,9 +514,7 @@ namespace core {
     {
       if (debug_sync)
       {
-        NV_IF_TARGET(NV_IS_HOST,
-                     (cudaStreamSynchronize(stream);),
-                     (cub::detail::device_synchronize();));
+        cub::DebugSyncStream(stream);
       }
     }