From 544e255ec0e048bcdca128464565d3e4977b3cde Mon Sep 17 00:00:00 2001 From: project6-dev Date: Thu, 13 Aug 2026 02:18:27 +0000 Subject: [PATCH] fix: copy_blocks use vllm_copy_cache --- qwen3_6_scripts/paged_attn.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/qwen3_6_scripts/paged_attn.py b/qwen3_6_scripts/paged_attn.py index e7daa2fe..e8a8f2c5 100644 --- a/qwen3_6_scripts/paged_attn.py +++ b/qwen3_6_scripts/paged_attn.py @@ -2256,4 +2256,5 @@ class PagedAttention: ) -> None: key_caches = [kv_cache[0] for kv_cache in kv_caches] value_caches = [kv_cache[1] for kv_cache in kv_caches] - ops.copy_blocks(key_caches, value_caches, src_to_dists) + import ixformer.functions as ixf_F + ixf_F.vllm_copy_cache(key_caches, value_caches, src_to_dists)