From 679f2c1e293d88ca2422940ca96715beeb85f2a4 Mon Sep 17 00:00:00 2001
From: root
 <root@cc-58e31738-7d16-4b93-aa09-745898b4ddee-0.cc-58e31738-7d16-4b93-aa09-745898b4ddee.ws-19872517457.svc.cluster.local>
Date: Wed, 19 Aug 2026 01:55:54 +0000
Subject: [PATCH] [fix] baseline4 _xllm_block_copy arg

---
 qwen3_6_scripts/paged_attn.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/qwen3_6_scripts/paged_attn.py b/qwen3_6_scripts/paged_attn.py
index 984ebf4a..ce5f10d5 100644
--- a/qwen3_6_scripts/paged_attn.py
+++ b/qwen3_6_scripts/paged_attn.py
@@ -2314,7 +2314,7 @@ class PagedAttention:
             dtype=torch.int64, device=device)
 
         numel_per_block = kv_caches[0][0][0].numel()
-        cache_dtype = kv_caches[0][0].dtype
+        cache_dtype = kv_caches[0][0].scalar_type()
 
         _xllm_block_copy(key_ptrs, val_ptrs,
                          src_indices, dst_indices, cum_sum,