change add pass

PaddlePaddle · Apr 17, 2024 · f878e99 · f878e99
1 parent c237db8
commit f878e99
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/test/ir/pir/fused_pass/onednn/test_operator_unsqueeze_onednn_fuse_pass.py b/test/ir/pir/fused_pass/onednn/test_operator_unsqueeze_onednn_fuse_pass.py
@@ -122,9 +122,9 @@ def build_ir_program(self):
                 transpose = paddle.transpose(squeeze_out, [0, 1, 2])
                 out = paddle.unsqueeze(transpose, [1])
                 out = paddle.assign(out)
-                self.pass_list = [
-                    'squeeze_transpose_onednn_fuse_pass',
-                    'operator_unsqueeze_onednn_fuse_pass',
+                self.pass_attr_list = [
+                    {'squeeze_transpose_onednn_fuse_pass': {}},
+                    {'operator_unsqueeze_onednn_fuse_pass': {}},
                 ]
                 self.feeds = {
                     "x": np.random.random((4, 16, 1, 32)).astype("float32"),
@@ -177,9 +177,9 @@ def build_ir_program(self):
                 relu_out = act_op(matmul)
                 out = paddle.unsqueeze(relu_out, [1])
                 out = paddle.assign(out)
-                self.pass_list = [
-                    'elementwise_act_onednn_fuse_pass',
-                    'operator_unsqueeze_onednn_fuse_pass',
+                self.pass_attr_list = [
+                    {'elementwise_act_onednn_fuse_pass': {}},
+                    {'operator_unsqueeze_onednn_fuse_pass': {}},
                 ]
                 self.feeds = {
                     "x": np.random.random((5, 5, 5, 5)).astype("float32"),