Skip to content

Commit 8f3c5f7

Browse files
committed
add linux specific logic
1 parent 094e841 commit 8f3c5f7

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

numba_cuda/numba/cuda/cuda_paths.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -314,10 +314,11 @@ def get_conda_include_dir():
314314

315315
def get_system_include_dir():
316316
"""Return the system CUDA include directory, if it exists"""
317-
system_cuda_include = '/usr/local/cuda/include'
318-
if os.path.exists(system_cuda_include):
319-
return system_cuda_include
320-
return None
317+
if sys.platform.startswith('linux'):
318+
system_cuda_include = '/usr/local/cuda/include'
319+
if os.path.exists(system_cuda_include):
320+
return system_cuda_include
321+
return
321322

322323

323324
def _get_include_dir():

0 commit comments

Comments
 (0)