From d0cd0a117f086fb36ea76e934bfda6ae3febcda4 Mon Sep 17 00:00:00 2001
From: feng_shuai <fengshuai03@baidu.com>
Date: Wed, 17 Aug 2022 14:24:51 +0800
Subject: [PATCH] fix:op version (#45192)

---
 paddle/fluid/framework/ir/vit_attention_fuse_pass.cc | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/paddle/fluid/framework/ir/vit_attention_fuse_pass.cc b/paddle/fluid/framework/ir/vit_attention_fuse_pass.cc
index 819708b447c..6591e1bbf12 100644
--- a/paddle/fluid/framework/ir/vit_attention_fuse_pass.cc
+++ b/paddle/fluid/framework/ir/vit_attention_fuse_pass.cc
@@ -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));
-- 
GitLab