cache_kernel -> cache_kernels

ffad4e1e · Woosuk Kwon · 6d2f74ef · ffad4e1e · ffad4e1e
Commit ffad4e1e authored Feb 16, 2023 by Woosuk Kwon
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

csrc/cache_kernels.cu csrc/cache_kernels.cu +0 -0

setup.py setup.py +1 -1

No files found.
--- a/csrc/cache_kernel.cu
+++ b/csrc/cache_kernel.cu
--- a/setup.py
+++ b/setup.py
@@ -10,7 +10,7 @@ ext_modules = []
 # Cache operations.
 cache_extension = cpp_extension.CUDAExtension(
    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},
 )
 ext_modules.append(cache_extension)