Use CUDA_PATH

lukstafi · May 8, 2024 · 89ab8c6 · 89ab8c6
1 parent 5051f7f
commit 89ab8c6
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 7 deletions.
diff --git a/cuda_ffi/dune b/cuda_ffi/dune
@@ -13,9 +13,9 @@
     (c_library_flags
      -lcuda
      -I
-     /usr/local/cuda/include
+     %{env:CUDA_PATH=/usr/local/cuda}/include
      -L
-     /usr/local/cuda/lib64/stubs)))
+     %{env:CUDA_PATH=/usr/local/cuda}/lib64/stubs)))
   (headers
    (include "cuda.h"))
   (type_description

diff --git a/debug.log b/debug.log
diff --git a/dune b/dune
@@ -1,7 +1,7 @@
 (env
  (_
-  (c_flags -I /usr/local/cuda/include)
-  (link_flags -cclib -L/usr/local/cuda/lib64/stubs)))
+  (c_flags -I %{env:CUDA_PATH=/usr/local/cuda}/include)
+  (link_flags -cclib -L%{env:CUDA_PATH=/usr/local/cuda}/lib64/stubs)))
 
 (library
  (public_name cudajit)

diff --git a/nvrtc_ffi/dune b/nvrtc_ffi/dune
@@ -13,9 +13,9 @@
     (c_library_flags
      -lnvrtc
      -I
-     /usr/local/cuda/include
+     %{env:CUDA_PATH=/usr/local/cuda}/include
      -L
-     /usr/local/cuda/lib64/stubs)))
+     %{env:CUDA_PATH=/usr/local/cuda}/lib64/stubs)))
   (headers
    (include "nvrtc.h"))
   (type_description