From ddd5a61311e059c209a19fac5430ee87c5f6d2ad Mon Sep 17 00:00:00 2001 From: achirkin Date: Fri, 3 Feb 2023 08:02:51 +0100 Subject: [PATCH] Revert #1224 --- cpp/include/raft/core/resource/cuda_stream.hpp | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) 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