modify setup.py

Desilo · Nov 23, 2023 · 67f66df · 67f66df
1 parent e51152f
commit 67f66df
Showing 1 changed file with 13 additions and 12 deletions.
diff --git a/setup.py b/setup.py
@@ -5,29 +5,29 @@
     CUDAExtension(
         name="randint_cuda",
         sources=[
-            "liberate/csprng/randint.cpp",
-            "liberate/csprng/randint_cuda_kernel.cu",
+            "src/liberate/csprng/randint.cpp",
+            "src/liberate/csprng/randint_cuda_kernel.cu",
         ],
     ),
     CUDAExtension(
         name="randround_cuda",
         sources=[
-            "liberate/csprng/randround.cpp",
-            "liberate/csprng/randround_cuda_kernel.cu",
+            "src/liberate/csprng/randround.cpp",
+            "src/liberate/csprng/randround_cuda_kernel.cu",
         ],
     ),
     CUDAExtension(
         name="discrete_gaussian_cuda",
         sources=[
-            "liberate/csprng/discrete_gaussian.cpp",
-            "liberate/csprng/discrete_gaussian_cuda_kernel.cu",
+            "src/liberate/csprng/discrete_gaussian.cpp",
+            "src/liberate/csprng/discrete_gaussian_cuda_kernel.cu",
         ],
     ),
     CUDAExtension(
         name="chacha20_cuda",
         sources=[
-            "liberate/csprng/chacha20.cpp",
-            "liberate/csprng/chacha20_cuda_kernel.cu",
+            "src/liberate/csprng/chacha20.cpp",
+            "src/liberate/csprng/chacha20_cuda_kernel.cu",
         ],
     ),
 ]
@@ -36,8 +36,8 @@
     CUDAExtension(
         name="ntt_cuda",
         sources=[
-            "liberate/ntt/ntt.cpp",
-            "liberate/ntt/ntt_cuda_kernel.cu",
+            "src/liberate/ntt/ntt.cpp",
+            "src/liberate/ntt/ntt_cuda_kernel.cu",
         ],
     )
 ]
@@ -50,7 +50,7 @@
         script_args=["build_ext"],
         options={
             "build": {
-                "build_lib": "liberate/csprng",
+                "build_lib": "src/liberate/csprng",
             }
         },
     )
@@ -62,7 +62,8 @@
         cmdclass={"build_ext": BuildExtension},
         options={
             "build": {
-                "build_lib": "liberate/ntt",
+                "build_lib": "src/liberate/ntt",
             }
         },
     )
+