From 6d2f74efb34566edc11477a651749173cac4c25d Mon Sep 17 00:00:00 2001
From: Woosuk Kwon <woosuk.kwon@berkeley.edu>
Date: Thu, 16 Feb 2023 09:24:42 +0000
Subject: [PATCH] Remove redundant fn

---
 csrc/cache.cpp | 9 +--------
 1 file changed, 1 insertion(+), 8 deletions(-)

diff --git a/csrc/cache.cpp b/csrc/cache.cpp
index 786452c0..ab33ee12 100644
--- 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");
 }