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

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