diff --git a/cpp/include/raft/core/resource/cuda_stream.hpp b/cpp/include/raft/core/resource/cuda_stream.hpp index fc69f10d83..6120a4b75c 100644 --- 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 \ No newline at end of file +} // namespace raft::resource