Remove redundant fn

6d2f74ef · Woosuk Kwon · 3b41f165 · 6d2f74ef
Commit 6d2f74ef authored Feb 16, 2023 by Woosuk Kwon
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 8 deletions

csrc/cache.cpp csrc/cache.cpp +1 -8

No files found.
--- a/csrc/cache.cpp
+++ b/csrc/cache.cpp
@@ -5,16 +5,9 @@ void copy_blocks(
  torch::Tensor& dst,
  const std::map<int64_t, int64_t>& block_mapping);
-void copy_cache_blocks(
-  torch::Tensor& src,
-  torch::Tensor& dst,
-  const std::map<int64_t, int64_t>& block_mapping) {
-  copy_blocks(src, dst, block_mapping);
-}
 PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
  m.def(
    "copy_cache_blocks",
-    &copy_cache_blocks,
+    &copy_blocks,
    "Copy the cache blocks from src to dst");
 }