From 49cd7def890841216b6d1bb274f991667240e8e4 Mon Sep 17 00:00:00 2001 From: claude Date: Fri, 14 Aug 2026 11:20:07 +0000 Subject: [PATCH] =?UTF-8?q?fix:=20slot=5Fids=20dtype=20long=E2=86=92int32?= =?UTF-8?q?=20in=20cache=20test?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- qwen3_6_scripts/verify_ast_chain.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/qwen3_6_scripts/verify_ast_chain.py b/qwen3_6_scripts/verify_ast_chain.py index 144ecdc3..9efa7784 100644 --- a/qwen3_6_scripts/verify_ast_chain.py +++ b/qwen3_6_scripts/verify_ast_chain.py @@ -160,7 +160,7 @@ def test_cache(): n_blocks = 8 block_size = 16 - slot_ids = torch.tensor([0, 1, 16, 17], dtype=torch.long, device="cuda") + slot_ids = torch.tensor([0, 1, 16, 17], dtype=torch.int32, device="cuda") keys = torch.randn(n_tokens, n_kv_heads, head_dim, dtype=torch.float16, device="cuda") values = torch.randn(n_tokens, n_kv_heads, head_dim, dtype=torch.float16, device="cuda") key_cache = torch.zeros(n_blocks, block_size, n_kv_heads, head_dim,