Use DebugSyncStream where appropriate.

NVIDIA · Jun 29, 2022 · cb30a6b · cb30a6b
1 parent 3b4d838
commit cb30a6b
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/thrust/system/cuda/detail/core/agent_launcher.h b/thrust/system/cuda/detail/core/agent_launcher.h
@@ -512,12 +512,7 @@ namespace core {
 
     THRUST_RUNTIME_FUNCTION void sync() const
     {
-      if (debug_sync)
-      {
-        NV_IF_TARGET(NV_IS_HOST,
-                     (cudaStreamSynchronize(stream);),
-                     (cub::detail::device_synchronize();));
-      }
+      CubDebug(cub::detail::DebugSyncStream(stream, this->debug_sync));
     }
 
     template<class K>