diff --git a/test/ir/inference/test_onednn_multi_gru_seq_fuse_pass.py b/test/ir/inference/test_onednn_multi_gru_seq_fuse_pass.py index 43a7f1952c8bd1..55ddd0d2490d4e 100644 --- a/test/ir/inference/test_onednn_multi_gru_seq_fuse_pass.py +++ b/test/ir/inference/test_onednn_multi_gru_seq_fuse_pass.py @@ -56,7 +56,7 @@ def generate_bias(shape): attrs={ 'origin_mode': False, 'is_reverse': False, - 'use_mkldnn': True, + 'use_onednn': True, }, ) @@ -75,7 +75,7 @@ def generate_bias(shape): attrs={ 'origin_mode': False, 'is_reverse': True, - 'use_mkldnn': True, + 'use_onednn': True, }, ) @@ -88,7 +88,7 @@ def generate_bias(shape): ] }, outputs={'Out': ['concat_output_1']}, - attrs={'axis': 1, 'use_mkldnn': True}, + attrs={'axis': 1, 'use_onednn': True}, ) fusion_gru_op_3 = OpConfig(