From 6e42a97367cfd1e4d8226ec4bee0d31ac91b85f4 Mon Sep 17 00:00:00 2001 From: Casper Date: Sat, 24 Feb 2024 17:20:28 +0100 Subject: [PATCH] Fix pybindings --- setup.py | 1 + 1 file changed, 1 insertion(+) diff --git a/setup.py b/setup.py index 35df839..f1e98be 100644 --- 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", ],