From 3e023929fb34ce559b84d8cfec37684fa83d2fe1 Mon Sep 17 00:00:00 2001 From: "Corey J. Nolet" Date: Wed, 1 Feb 2023 18:21:22 -0500 Subject: [PATCH 1/2] Removing interruptible from basic handle sync. --- cpp/include/raft/core/resource/cuda_stream.hpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cpp/include/raft/core/resource/cuda_stream.hpp b/cpp/include/raft/core/resource/cuda_stream.hpp index 318252199e..db95ff5000 100644 --- a/cpp/include/raft/core/resource/cuda_stream.hpp +++ b/cpp/include/raft/core/resource/cuda_stream.hpp @@ -91,7 +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) { - interruptible::synchronize(stream); + RAFT_CUDA_TRY(cudaStreamSynchronize(stream)); } /** From 73c026c2a428283723c077b390cbd6702aff3996 Mon Sep 17 00:00:00 2001 From: "Corey J. Nolet" Date: Wed, 1 Feb 2023 18:34:06 -0500 Subject: [PATCH 2/2] Adding todo and referencing github issue --- cpp/include/raft/core/resource/cuda_stream.hpp | 3 +++ 1 file changed, 3 insertions(+) diff --git a/cpp/include/raft/core/resource/cuda_stream.hpp b/cpp/include/raft/core/resource/cuda_stream.hpp index db95ff5000..fc69f10d83 100644 --- a/cpp/include/raft/core/resource/cuda_stream.hpp +++ b/cpp/include/raft/core/resource/cuda_stream.hpp @@ -91,6 +91,9 @@ 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)); }