Rename awq_cuda -> awq_ext

3a85bf4c · Casper · 98f700d7 · 3a85bf4c
Commit 3a85bf4c authored Dec 22, 2023 by Casper
Hide whitespace changes
Inline Side-by-side

Showing with 8 additions and 8 deletions

setup.py setup.py +8 -8

No files found.
--- a/setup.py
+++ b/setup.py
@@ -122,11 +122,11 @@ extensions = [
    CUDAExtension(
        "awq_ext",
        [
-            "awq_cuda/pybind_awq.cpp",
-            "awq_cuda/quantization/gemm_cuda_gen.cu",
-            "awq_cuda/layernorm/layernorm.cu",
-            "awq_cuda/position_embedding/pos_encoding_kernels.cu",
-            "awq_cuda/quantization/gemv_cuda.cu"
+            "awq_ext/pybind_awq.cpp",
+            "awq_ext/quantization/gemm_cuda_gen.cu",
+            "awq_ext/layernorm/layernorm.cu",
+            "awq_ext/position_embedding/pos_encoding_kernels.cu",
+            "awq_ext/quantization/gemv_cuda.cu"
        ], extra_compile_args=extra_compile_args
    )
 ]
@@ -136,9 +136,9 @@ if os.name != "nt":
        CUDAExtension(
            "awq_ft_ext",
            [
-                "awq_cuda/pybind_awq_ft.cpp",
-                "awq_cuda/attention/ft_attention.cpp",
-                "awq_cuda/attention/decoder_masked_multihead_attention.cu"
+                "awq_ext/pybind_awq_ft.cpp",
+                "awq_ext/attention/ft_attention.cpp",
+                "awq_ext/attention/decoder_masked_multihead_attention.cu"
            ], extra_compile_args=extra_compile_args
        )
    )