From 6aee9b7ac7c97181844227a5ef3d7dc4d0fed922 Mon Sep 17 00:00:00 2001
From: HighCloud <highcloud333@outlook.com>
Date: Tue, 19 Aug 2025 14:24:49 +0800
Subject: [PATCH] deepseek mcore support 310P

---
 vllm_mindspore/model_executor/models/model_base.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/vllm_mindspore/model_executor/models/model_base.py b/vllm_mindspore/model_executor/models/model_base.py
index 421e039c5..e877479f8 100644
--- a/vllm_mindspore/model_executor/models/model_base.py
+++ b/vllm_mindspore/model_executor/models/model_base.py
@@ -78,7 +78,7 @@ class MLAAttentionWrapper(AttentionWrapper):
         if not self.use_ringmla:
             self.kv_cache = [
                 (
-                    ms.mint.zeros(
+                    create_kv_cache(
                         self.kv_shape,  # type: ignore[misc]
                         dtype=vllm_config.model_config.dtype), ) for _ in
                 range(vllm_config.parallel_config.pipeline_parallel_size)
-- 
Gitee