diff --git a/mindspore-lite/test/config_level0/models_cloud_ascend_a2.cfg b/mindspore-lite/test/config_level0/models_cloud_ascend_a2.cfg index 75488ee1fdb7c1e5464083c368e79f4789c027f3..4cd5b8ecbfe70419f64004a40af7fdcc6c4e3884 100644 --- a/mindspore-lite/test/config_level0/models_cloud_ascend_a2.cfg +++ b/mindspore-lite/test/config_level0/models_cloud_ascend_a2.cfg @@ -14,4 +14,5 @@ single_op_gns.onnx;3:x,gamma,beta;1,192,160,160:192:192;static;; 1 long_sequence_eta.pb;2:id,wt;1,576:1,576;static;; 1 vod_sr_M5_H10_20231107_manualv4_int8_dynamic.onnx;1:input1;1,3,1080,1920;static;; 2 user_latent_vector.pb;3:embedding-feature,dense,emb;1,10688:1,1:1,128;static;; 1 -single_matmul_model.onnx;1:input;1,4;static;; 1 \ No newline at end of file +single_matmul_model.onnx;1:input;1,4;static;; 1 +fas.mindir;3:query,key,value;1,12,287,128:1,12,287,128:1,12,287,128;static;; 1 \ No newline at end of file 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 63f14d2b5e0f591bac3751f01196b2b7d6df1394..a04c0663035a682c73b7f3b1bbb5665652a6d64f 100644 --- a/mindspore-lite/tools/optimizer/graph/attr_to_args_pass.cc +++ b/mindspore-lite/tools/optimizer/graph/attr_to_args_pass.cc @@ -40,6 +40,15 @@ static const std::map>> {"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}}},