Skip to content

Commit f854461

Browse files
co63ocmaxiaolong001
authored andcommitted
test_mkldnn_matmul_v2_elementwise_add_fuse_pass.py modify use_mkldnn [fluid_ops] (PaddlePaddle#74507)
* Fix * Fix
1 parent eabb554 commit f854461

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/ir/inference/test_mkldnn_matmul_v2_elementwise_add_fuse_pass.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ def generate_input(type):
5353
type='matmul_v2',
5454
inputs={'X': ['matmul_X'], 'Y': ['matmul_Y']},
5555
outputs={'Out': ['matmul_output']},
56-
attrs={'use_mkldnn': True},
56+
attrs={'use_onednn': True},
5757
)
5858

5959
if matmul_as_x:
@@ -65,7 +65,7 @@ def generate_input(type):
6565
type='elementwise_add',
6666
inputs=inputs,
6767
outputs={'Out': ['elementwise_add_output']},
68-
attrs={'axis': axis, 'use_mkldnn': True},
68+
attrs={'axis': axis, 'use_onednn': True},
6969
)
7070

7171
model_net = [matmul_op, elt_add_op]

0 commit comments

Comments
 (0)