fix(python): JIT compilation on certain CUDA versions

Tom94 · Tom94 · commit dc5c9e76bd35 · 2025-07-08T17:48:46.000+02:00
diff --git a/bindings/torch/setup.py b/bindings/torch/setup.py
@@ -177,7 +177,7 @@ def find_cl_path():
 else:
 	cuda_include_dir = os.path.join(os.path.dirname(os.path.dirname(nvcc_path)), "include")
 
-	cuda_headers = glob(f"{cuda_include_dir}/cuda_fp16*")
+	cuda_headers = glob(f"{cuda_include_dir}/cuda_fp16*") + glob(f"{cuda_include_dir}/vector*")
 	tcnn_headers = glob(f"{root_dir}/include/tiny-cuda-nn/*", recursive=True)
 	pcg32_headers = glob(f"{root_dir}/dependencies/pcg32/*")
 
