From d156f6bd47bbb9de5e8f0d81412dd930d0129af1 Mon Sep 17 00:00:00 2001
From: yihangchen <chenyihang5@huawei.com>
Date: Mon, 14 Jul 2025 16:21:00 +0800
Subject: [PATCH] flash attention score attr map

---
 .../tools/optimizer/graph/attr_to_args_pass.cc           | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/mindspore-lite/tools/optimizer/graph/attr_to_args_pass.cc b/mindspore-lite/tools/optimizer/graph/attr_to_args_pass.cc
index fd6679f4..78862f54 100644
--- a/mindspore-lite/tools/optimizer/graph/attr_to_args_pass.cc
+++ b/mindspore-lite/tools/optimizer/graph/attr_to_args_pass.cc
@@ -39,6 +39,15 @@ static const std::map<std::string, std::vector<std::pair<std::string, size_t>>>
     {"num_key_value_heads", 18},
     {"sparse_mode", 19},
     {"inner_precise", 20}}},
+  {"FlashAttentionScore",
+   {{"head_num", 11},
+    {"keep_prob", 12},
+    {"scale_value", 13},
+    {"pre_tockens", 14},
+    {"next_tockens", 15},
+    {"inner_precise", 16},
+    {"input_layout", 17},
+    {"sparse_mode", 18}}},
   {"BroadcastTo", {{"shape", 2}}},
   {"ArgMaxV2", {{"axis", 2}, {"output_type", 3}}},
   {"ArgMaxWithValue", {{"axis", 2}, {"keep_dims", 3}}},
-- 
Gitee