From ae8a7bcf51551e95a1ee5cd206d842a9b960924f Mon Sep 17 00:00:00 2001
From: jakirkham <jakirkham@gmail.com>
Date: Wed, 6 Mar 2024 13:54:59 -0800
Subject: [PATCH] Drop version hard-coding from YCM conf

---
 .ycm_extra_conf.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/.ycm_extra_conf.py b/.ycm_extra_conf.py
index f14a0da5..203c5058 100644
--- a/.ycm_extra_conf.py
+++ b/.ycm_extra_conf.py
@@ -5,13 +5,13 @@
 
 CONDA_PREFIX = os.environ["CONDA_PREFIX"]
 NVJITLINK_DIR = Path(__file__).parent
-VER = sys.version_info
-PYTHON_INCLUDE_NAME = "python3.10"  # Not ideal
+PYTHON_VERSION = sys.version_info
+PYTHON_INCLUDE_NAME = "python%d.%d" % PYTHON_VERSION
 
 CONDA_INCLUDE_DIR = Path(CONDA_PREFIX, "include")
 PYTHON_INCLUDE_DIR = Path(CONDA_INCLUDE_DIR, PYTHON_INCLUDE_NAME)
 NVJITLINK_INCLUDE_DIR = Path(NVJITLINK_DIR, "include")
-CUDA_INCLUDE_DIR = "/usr/local/cuda-12.0/include"
+CUDA_INCLUDE_DIR = "/usr/local/cuda/include"
 
 flags = [
     "--cuda-gpu-arch=sm_50",