cuda+py: move C extension into cufinufft package

Instead of living in the site-packages root, we put this in the `cufinufft` package, similarly to `finufft`.
janden · Dec 18, 2023 · 1515186 · 1515186
1 parent baa71b8
commit 1515186
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/python/cufinufft/cufinufft/_cufinufft.py b/python/cufinufft/cufinufft/_cufinufft.py
@@ -42,7 +42,7 @@
 try:
     if lib is None:
         # Find the library.
-        fh = imp.find_module('cufinufftc')[0]
+        fh = imp.find_module('cufinufft/cufinufftc')[0]
         # Get the full path for the ctypes loader.
         full_lib_path = os.path.realpath(fh.name)
         fh.close()    # Be nice and close the open file handle.

diff --git a/python/cufinufft/setup.py b/python/cufinufft/setup.py
@@ -68,9 +68,9 @@
     #   that is rpath linked to CUDA library, also decorated (by auditwheel).
     #   Most importantly, pip will manage to install all this stuff in
     #   in places Python can find it (with a little help).
-    py_modules=['cufinufftc'],
+    py_modules=['cufinufft.cufinufftc'],
     ext_modules=[
-        Extension(name='cufinufftc',
+        Extension(name='cufinufft.cufinufftc',
                   sources=[],
                   libraries=['cufinufft'],
                   library_dirs=[library_dir],