Fix pybindings

casper-hansen · Feb 24, 2024 · 6e42a97 · 6e42a97
1 parent 1d3808a
commit 6e42a97
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -189,6 +189,7 @@ def get_extra_link_args():
         CUDAExtension(
             "awq_v2_ext",
             [
+                "awq_ext/pybind_awq_v2.cpp",
                 "awq_ext/quantization_new/gemv/gemv_cuda.cu",
                 "awq_ext/quantization_new/gemm/gemm_cuda.cu",
             ],