Skip to content

Commit 1a440f0

Browse files
co63ocmaxiaolong001
authored andcommitted
test_onednn_multi_gru_seq_fuse_pass.py modify use_mkldnn [fluid_ops] (PaddlePaddle#74509)
1 parent 10bf1ec commit 1a440f0

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

test/ir/inference/test_onednn_multi_gru_seq_fuse_pass.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ def generate_bias(shape):
5656
attrs={
5757
'origin_mode': False,
5858
'is_reverse': False,
59-
'use_mkldnn': True,
59+
'use_onednn': True,
6060
},
6161
)
6262

@@ -75,7 +75,7 @@ def generate_bias(shape):
7575
attrs={
7676
'origin_mode': False,
7777
'is_reverse': True,
78-
'use_mkldnn': True,
78+
'use_onednn': True,
7979
},
8080
)
8181

@@ -88,7 +88,7 @@ def generate_bias(shape):
8888
]
8989
},
9090
outputs={'Out': ['concat_output_1']},
91-
attrs={'axis': 1, 'use_mkldnn': True},
91+
attrs={'axis': 1, 'use_onednn': True},
9292
)
9393

9494
fusion_gru_op_3 = OpConfig(

0 commit comments

Comments
 (0)