From d4db9b028b865c187969402289fb9cfe5dabbf54 Mon Sep 17 00:00:00 2001 From: shilinlee <836160610@qq.com> Date: Fri, 15 Aug 2025 14:58:42 +0800 Subject: [PATCH] fix: the store_dtype typo for ascend mla (#9208) Signed-off-by: shilinlee_com <836160610@qq.com> --- python/sglang/srt/mem_cache/memory_pool.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/sglang/srt/mem_cache/memory_pool.py b/python/sglang/srt/mem_cache/memory_pool.py index cc3faea0a..3b567dc76 100644 --- a/python/sglang/srt/mem_cache/memory_pool.py +++ b/python/sglang/srt/mem_cache/memory_pool.py @@ -951,7 +951,7 @@ class AscendMLAPagedTokenToKVPool(MLATokenToKVPool): cache_k = cache_k.to(self.dtype) if self.store_dtype != self.dtype: - cache_k = cache_k.view(store_dtype) + cache_k = cache_k.view(self.store_dtype) import torch_npu