diff --git a/paddle/fluid/framework/ir/vit_attention_fuse_pass.cc b/paddle/fluid/framework/ir/vit_attention_fuse_pass.cc
index 819708b447cdb5d26d973f3a33cfcc356751e416..6591e1bbf12d06c360f76057420de2c08d83e03c 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));