Revert rapidsai#1224

achirkin · Feb 3, 2023 · ddd5a61 · ddd5a61
1 parent 456e8e2
commit ddd5a61
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/cpp/include/raft/core/resource/cuda_stream.hpp b/cpp/include/raft/core/resource/cuda_stream.hpp
@@ -91,10 +91,7 @@ inline void set_cuda_stream(resources const& res, rmm::cuda_stream_view stream_v
  */
 inline void sync_stream(const resources& res, rmm::cuda_stream_view stream)
 {
-  // TODO: Fix interruptible segfault:
-  // https://github.com/rapidsai/raft/issues/1225
-  // interruptible::synchronize(stream);
-  RAFT_CUDA_TRY(cudaStreamSynchronize(stream));
+  interruptible::synchronize(stream);
 }
 
 /**
@@ -106,4 +103,4 @@ inline void sync_stream(const resources& res) { sync_stream(res, get_cuda_stream
  * @}
  */
 
-}  // namespace raft::resource
+}  // namespace raft::resource