Rename exllama extensions (#4)

4bb0c022 · Casper · GitHub · e6398f88 · 4bb0c022
Unverified Commit 4bb0c022 authored Jan 21, 2024 by Casper Committed by GitHub Jan 21, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

setup.py setup.py +2 -2

No files found.
--- a/setup.py
+++ b/setup.py
@@ -158,7 +158,7 @@ extensions = [

 extensions.append(
    CUDAExtension(
-        "exllama_kernels",
+        "exl_ext",
        [
            "awq_ext/exllama/exllama_ext.cpp",
            "awq_ext/exllama/cuda_buffers.cu",
@@ -172,7 +172,7 @@ extensions.append(
 )
 extensions.append(
    CUDAExtension(
-        "exllamav2_kernels",
+        "exlv2_ext",
        [
            "awq_ext/exllamav2/ext.cpp",
            "awq_ext/exllamav2/cuda/q_matrix.cu",