diff --git a/cpp/test/sparse/csr_row_slice.cu b/cpp/test/sparse/csr_row_slice.cu index c8c593790f..e37827d18d 100644 --- a/cpp/test/sparse/csr_row_slice.cu +++ b/cpp/test/sparse/csr_row_slice.cu @@ -124,7 +124,7 @@ class CSRRowSliceTest : public ::testing::TestWithParam out_ref_h = params.out_ref_h; update_device(out_ref.data(), out_ref_h.data(), out_ref_h.size(), stream); - RAFT_CHECK_CUDA(cudaStreamSynchronize(stream)); + RAFT_CUDA_TRY(cudaStreamSynchronize(stream)); } void SetUp() override