add csrc

049da596 · nicodafagood · 49e84bec · 049da596
Commit 049da596 authored Sep 13, 2024 by nicodafagood
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

setup.py setup.py +1 -0

No files found.
--- a/setup.py
+++ b/setup.py
@@ -339,6 +339,7 @@ vllm_extension_sources = [
    "csrc/layernorm_kernels.cu",
    "csrc/quantization/squeezellm/quant_cuda_kernel.cu",
    "csrc/quantization/gptq/q_gemm.cu",
+    "csrc/quantization/myq/q_gemm.cu",
    "csrc/cuda_utils_kernels.cu",
    "csrc/moe_align_block_size_kernels.cu",
    "csrc/pybind.cpp",