Use driver API

2025-08-19 11:29:15 +08:00
parent 51d1e9cdd3
commit ea9c5d9270
2 changed files with 4 additions and 4 deletions
--- a/csrc/jit/handle.hpp
+++ b/csrc/jit/handle.hpp
@@ -8,7 +8,7 @@

 namespace deep_gemm {

-#if CUDART_VERSION >= 12080 and not defined(DG_JIT_USE_DRIVER_API)
+#if CUDART_VERSION >= 12080 and defined(DG_JIT_USE_RUNTIME_API)

 // Use CUDA runtime API
 using LibraryHandle = cudaLibrary_t;