zijie-tian/nano-vllm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[fix] Fixed compile problem.

Browse Source
This commit is contained in:
Zijie Tian
2025-12-26 21:02:43 +08:00
parent 82ed34fc2d
commit 600af0f59c
1 changed files with 3 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
setup.py
View File

@@ -1,9 +1,5 @@
from setuptools import setup, find_packages from setuptools import setup, find_packages
from torch.utils.cpp_extension import BuildExtension, CUDAExtension 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( setup(
name='nano-vllm', name='nano-vllm',
@@ -15,15 +11,15 @@ setup(
CUDAExtension( CUDAExtension(
name='nanovllm.comm._sgdma_cuda', name='nanovllm.comm._sgdma_cuda',
sources=[ sources=[
os.path.join(project_root, 'csrc', 'sgdma.cpp'), 'csrc/sgdma.cpp',
os.path.join(project_root, 'csrc', 'sgdma_kernel.cu'), 'csrc/sgdma_kernel.cu',
], ],
extra_compile_args={ extra_compile_args={
'cxx': ['-O3', '-std=c++17'], 'cxx': ['-O3', '-std=c++17'],
'nvcc': ['-O3', '--use_fast_math', '-std=c++17'] 'nvcc': ['-O3', '--use_fast_math', '-std=c++17']
}, },
include_dirs=[ include_dirs=[
os.path.join(project_root, 'csrc'), 'csrc',
], ],
) )
], ],