[CLEANUP]

kyegomez · Mar 25, 2024 · 171f4e5 · 171f4e5
1 parent 138fa57
commit 171f4e5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -125,7 +125,7 @@ def append_nvcc_threads(nvcc_extra_args):
 
     ext_modules.append(
         CUDAExtension(
-            name="selective_scan_cuda",
+            name="gemm_lowbit_ext",
             sources=[
                 "kernel/gemm_lowbit_kernel.cu",
 
@@ -151,7 +151,7 @@ def append_nvcc_threads(nvcc_extra_args):
                     + cc_flag
                 ),
             },
-            include_dirs=[Path(this_dir) / "csrc" / "selective_scan"],
+            include_dirs=[Path(this_dir) / "kernel" / "bitnet"],
         )
     )