Add back CUDA_INCLUDE_DIR in YCM conf

rapidsai · Mar 7, 2024 · 1cfd711 · 1cfd711
1 parent 38d1d56
commit 1cfd711
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/.ycm_extra_conf.py b/.ycm_extra_conf.py
@@ -7,12 +7,14 @@
 PREFIX_INCLUDE_DIR = Path(PREFIX, "include")
 PYTHON_INCLUDE_DIR = sysconfig.get_config_var("include")
 NVJITLINK_INCLUDE_DIR = Path(NVJITLINK_DIR, "include")
+CUDA_INCLUDE_DIR = "/usr/local/cuda/include"
 
 flags = [
     "--cuda-gpu-arch=sm_50",
     f"-I{PREFIX_INCLUDE_DIR}",
     f"-I{PYTHON_INCLUDE_DIR}",
     f"-I{NVJITLINK_INCLUDE_DIR}",
+    f"-I{CUDA_INCLUDE_DIR}",
 ]