Add to setup

casper-hansen · Feb 24, 2024 · 5e2913a · 5e2913a
1 parent d9ab410
commit 5e2913a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -175,6 +175,8 @@ def get_extra_link_args():
                 "awq_ext/vllm/moe_alig_block.cu",
                 "awq_ext/vllm/activation.cu",
                 "awq_ext/vllm/topk_softmax_kernels.cu",
+                "awq_ext/quantization_new/gemv/gemv_cuda.cu",
+                "awq_ext/quantization_new/gemm/gemm_cuda.cu",
             ],
             extra_compile_args=extra_compile_args,
         )