From 29b977fa1e3d48ff60772ad7cf4920f33172a250 Mon Sep 17 00:00:00 2001
From: osieosie <jyin27@jh.edu>
Date: Tue, 21 Nov 2023 03:21:59 -0500
Subject: [PATCH] Update setup.py

Fixed quant_cuda import error
---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index bd610bc..b7fbdc8 100644
--- a/setup.py
+++ b/setup.py
@@ -13,7 +13,7 @@
 setup(
     name='quant_cuda',
     ext_modules=[cpp_extension.CUDAExtension(
-        'llmtune.engine.quant.quant_cuda',
+        'quant_cuda', 
         [
         	'llmtools/engine/inference/cuda/quant_cuda.cpp', 
         	'llmtools/engine/inference/cuda/quant_cuda_kernel.cu'