diff --git a/test/ir/inference/test_mkldnn_matmul_v2_elementwise_add_fuse_pass.py b/test/ir/inference/test_mkldnn_matmul_v2_elementwise_add_fuse_pass.py index d6be1efaa34353..cf383495f52c42 100644 --- a/test/ir/inference/test_mkldnn_matmul_v2_elementwise_add_fuse_pass.py +++ b/test/ir/inference/test_mkldnn_matmul_v2_elementwise_add_fuse_pass.py @@ -53,7 +53,7 @@ def generate_input(type): type='matmul_v2', inputs={'X': ['matmul_X'], 'Y': ['matmul_Y']}, outputs={'Out': ['matmul_output']}, - attrs={'use_mkldnn': True}, + attrs={'use_onednn': True}, ) if matmul_as_x: @@ -65,7 +65,7 @@ def generate_input(type): type='elementwise_add', inputs=inputs, outputs={'Out': ['elementwise_add_output']}, - attrs={'axis': axis, 'use_mkldnn': True}, + attrs={'axis': axis, 'use_onednn': True}, ) model_net = [matmul_op, elt_add_op]