fix: the store_dtype typo for ascend mla (#9208)

Signed-off-by: shilinlee_com <836160610@qq.com>
2025-08-15 14:58:42 +08:00
parent f7dd651dbd
commit d4db9b028b
1 changed files with 1 additions and 1 deletions
--- 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