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

Remove redundant fn

Browse Source
This commit is contained in:
Woosuk Kwon 2023-02-16 09:24:42 +00:00
parent 3b41f16596
commit 6d2f74efb3
1 changed files with 1 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
csrc/cache.cpp
View File

@ -5,16 +5,9 @@ void copy_blocks(
torch::Tensor& dst, torch::Tensor& dst,
const std::map<int64_t, int64_t>& block_mapping); 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) { PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
m.def( m.def(
"copy_cache_blocks", "copy_cache_blocks",
&copy_cache_blocks, &copy_blocks,
"Copy the cache blocks from src to dst"); "Copy the cache blocks from src to dst");
} }