fix cuda build missing CUDASUCCESS_OR_FATAL in vector_cudahost.h

9655896f · Jeff Daily · 3eafff74 · 9655896f
Commit 9655896f authored Sep 23, 2025 by Jeff Daily
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 3 deletions

include/LightGBM/cuda/vector_cudahost.h include/LightGBM/cuda/vector_cudahost.h +2 -3

No files found.
--- a/include/LightGBM/cuda/vector_cudahost.h
+++ b/include/LightGBM/cuda/vector_cudahost.h
@@ -9,12 +9,11 @@
 #include <LightGBM/utils/common.h>

 #ifdef USE_CUDA
-#if defined(USE_ROCM)
-#include <LightGBM/cuda/cuda_utils.hu>
-#else  // USE_ROCM
+#ifndef USE_ROCM
 #include <cuda.h>
 #include <cuda_runtime.h>
 #endif  // USE_ROCM
+#include <LightGBM/cuda/cuda_utils.hu>
 #endif  // USE_CUDA
 #include <stdio.h>