Fix compilation

rapidsai · Sep 11, 2024 · e76eb18 · e76eb18
1 parent 9175b86
commit e76eb18
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/cpp/include/raft/linalg/detail/eig.cuh b/cpp/include/raft/linalg/detail/eig.cuh
@@ -98,11 +98,12 @@ void eigDC(raft::resources const& handle,
   int cudart_version = 0;
   RAFT_CUDA_TRY(cudaRuntimeGetVersion(&cudart_version));
   cudaStream_t stream_new;
+  cudaEvent_t sync_event;
+  rmm::cuda_stream stream_new_wrapper;
   if (cudart_version < 12050) {
     // Use a new stream instead of `cudaStreamPerThread` to avoid cusolver bug # 4580093.
-    rmm::cuda_stream stream_new_wrapper;
-    stream_new             = stream_new_wrapper.value();
-    cudaEvent_t sync_event = resource::detail::get_cuda_stream_sync_event(handle);
+    stream_new = stream_new_wrapper.value();
+    sync_event = resource::detail::get_cuda_stream_sync_event(handle);
     RAFT_CUDA_TRY(cudaEventRecord(sync_event, stream));
     RAFT_CUDA_TRY(cudaStreamWaitEvent(stream_new, sync_event));
   } else {