Fix nvcc compilation (#316)

ratt-ru · Oct 28, 2024 · 7b1526f · 7b1526f
1 parent 3187370
commit 7b1526f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/HISTORY.rst b/HISTORY.rst
@@ -4,6 +4,7 @@ History
 
 X.Y.Z (YYYY-MM-DD)
 ------------------
+* Fix nvcc compilation (:pr:`316`)
 * Workaround numba #5929 (:pr:`312`)
 * Restrict NumPy to less than 2.0.0 (:pr:`313`)
 

diff --git a/africanus/util/nvcc.py b/africanus/util/nvcc.py
@@ -14,10 +14,10 @@
 
 from os.path import join as pjoin
 
-import distutils
-from distutils import errors
-from distutils import msvccompiler
-from distutils import unixccompiler
+import setuptools
+from setuptools import errors
+import setuptools._distutils._msvccompiler as msvccompiler
+import setuptools._distutils.unixccompiler as unixccompiler
 
 from africanus.util.code import format_code
 from africanus.util.requirements import requires_optional
@@ -106,7 +106,7 @@ def get_cuda_path():
 def get_nvcc_path():
     nvcc = os.environ.get("NVCC", None)
     if nvcc:
-        return distutils.split_quoted(nvcc)
+        return setuptools.split_quoted(nvcc)
 
     cuda_path = get_cuda_path()
     if cuda_path is None: