未验证 提交 d0cd0a11 编写于 作者: F feng_shuai 提交者: GitHub

fix:op version (#45192)

上级 cccba68c
......@@ -139,9 +139,9 @@ REGISTER_PASS(vit_attention_fuse_pass,
REGISTER_PASS_CAPABILITY(vit_attention_fuse_pass)
.AddCombination(
paddle::framework::compatible::OpVersionComparatorCombination()
.GE("reshape2", 0)
.GE("transpose2", 0)
.GE("slice", 0)
.GE("scale", 0)
.GE("softmax", 0)
.GE("matmul_v2", 0));
.EQ("reshape2", 0)
.EQ("transpose2", 0)
.EQ("slice", 0)
.EQ("scale", 0)
.EQ("softmax", 0)
.EQ("matmul_v2", 0));
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册