From 600af0f59c9bd1e1df917fcbf8d228b16765dee7 Mon Sep 17 00:00:00 2001
From: Zijie Tian <zijietian@mail.xmu.edu.cn>
Date: Fri, 26 Dec 2025 21:02:43 +0800
Subject: [PATCH] [fix] Fixed compile problem.

---
 setup.py | 10 +++-------
 1 file changed, 3 insertions(+), 7 deletions(-)

diff --git a/setup.py b/setup.py
index 1391997..7473440 100644
--- a/setup.py
+++ b/setup.py
@@ -1,9 +1,5 @@
 from setuptools import setup, find_packages
 from torch.utils.cpp_extension import BuildExtension, CUDAExtension
-import os
-
-# Get the absolute path to the project root
-project_root = os.path.dirname(os.path.abspath(__file__))
 
 setup(
     name='nano-vllm',
@@ -15,15 +11,15 @@ setup(
         CUDAExtension(
             name='nanovllm.comm._sgdma_cuda',
             sources=[
-                os.path.join(project_root, 'csrc', 'sgdma.cpp'),
-                os.path.join(project_root, 'csrc', 'sgdma_kernel.cu'),
+                'csrc/sgdma.cpp',
+                'csrc/sgdma_kernel.cu',
             ],
             extra_compile_args={
                 'cxx': ['-O3', '-std=c++17'],
                 'nvcc': ['-O3', '--use_fast_math', '-std=c++17']
             },
             include_dirs=[
-                os.path.join(project_root, 'csrc'),
+                'csrc',
             ],
         )
     ],