Runtime check CUDA driver version to avoid unresolved green context symbols (#9021)

2025-08-12 09:26:10 -07:00
parent 3a9afe2a42
commit 445f9dca6e
3 changed files with 28 additions and 36 deletions
--- a/sgl-kernel/csrc/spatial/cuda_utils.h
+++ b/sgl-kernel/csrc/spatial/cuda_utils.h
@@ -1,6 +1,8 @@
 #include <cuda.h>
 #include <cuda_runtime.h>

+#include <iostream>
+
 #define CUDA_RT(call)                                                                                        \
  do {                                                                                                       \
    cudaError_t _status = (call);                                                                            \