20231088/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

cache_kernel -> cache_kernels

Browse Source
This commit is contained in:
Woosuk Kwon 2023-02-16 20:05:45 +00:00
parent 6d2f74efb3
commit ffad4e1e03
2 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

0
csrc/cache_kernel.cu → csrc/cache_kernels.cu
View File

2
setup.py
View File

@ -10,7 +10,7 @@ ext_modules = []
# Cache operations. # Cache operations.
cache_extension = cpp_extension.CUDAExtension( cache_extension = cpp_extension.CUDAExtension(
name='cacheflow.ops', name='cacheflow.ops',
sources=['csrc/cache.cpp', 'csrc/cache_kernel.cu'], sources=['csrc/cache.cpp', 'csrc/cache_kernels.cu'],
extra_compile_args={'cxx': CXX_FLAGS, 'nvcc': NVCC_FLAGS}, extra_compile_args={'cxx': CXX_FLAGS, 'nvcc': NVCC_FLAGS},
) )
ext_modules.append(cache_extension) ext_modules.append(cache_extension)