cache_kernel -> cache_kernels

2023-02-16 20:05:45 +00:00 · 2023-02-16 20:05:45 +00:00 · ffad4e1e03
commit ffad4e1e03
parent 6d2f74efb3
2 changed files with 1 additions and 1 deletions
--- a/csrc/cache_kernels.cu
+++ b/csrc/cache_kernels.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)