diff --git a/cpp/include/raft/core/resource/cuda_stream.hpp b/cpp/include/raft/core/resource/cuda_stream.hpp index 318252199e..fc69f10d83 100644 --- a/cpp/include/raft/core/resource/cuda_stream.hpp +++ b/cpp/include/raft/core/resource/cuda_stream.hpp @@ -91,7 +91,10 @@ 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); + // TODO: Fix interruptible segfault: + // https://github.com/rapidsai/raft/issues/1225 + // interruptible::synchronize(stream); + RAFT_CUDA_TRY(cudaStreamSynchronize(stream)); } /**