From 4bb0c022e953e28ee0f27688f62467594682437e Mon Sep 17 00:00:00 2001
From: Casper <casperbh.96@gmail.com>
Date: Sun, 21 Jan 2024 13:25:08 +0100
Subject: [PATCH] Rename exllama extensions (#4)

---
 setup.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index f38ec49..94cc778 100644
--- a/setup.py
+++ b/setup.py
@@ -158,7 +158,7 @@ def get_compute_capabilities():
 
 extensions.append(
     CUDAExtension(
-        "exllama_kernels",
+        "exl_ext",
         [
             "awq_ext/exllama/exllama_ext.cpp",
             "awq_ext/exllama/cuda_buffers.cu",
@@ -172,7 +172,7 @@ def get_compute_capabilities():
 )
 extensions.append(
     CUDAExtension(
-        "exllamav2_kernels",
+        "exlv2_ext",
         [
             "awq_ext/exllamav2/ext.cpp",
             "awq_ext/exllamav2/cuda/q_matrix.cu",