Skip to content

Commit 298c9d4

Browse files
authored
test_conv_transpose_bn_fuse_pass.py modify use_mkldnn [fluid_ops] (#74535)
1 parent b294533 commit 298c9d4

4 files changed

+8
-8
lines changed

test/ir/inference/test_conv_transpose_bn_fuse_pass.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -133,7 +133,7 @@ def generate_batch_norm_Variance():
133133
'data_format': random_data_layout,
134134
'output_size': random_output_size,
135135
'output_padding': random_output_size,
136-
'use_mkldnn': random_use_onednn,
136+
'use_onednn': random_use_onednn,
137137
'is_test': True,
138138
},
139139
)
@@ -160,7 +160,7 @@ def generate_batch_norm_Variance():
160160
'is_test': True,
161161
'trainable_statistics': False,
162162
'data_layout': random_data_layout,
163-
'use_mkldnn': random_use_onednn,
163+
'use_onednn': random_use_onednn,
164164
},
165165
)
166166

@@ -194,7 +194,7 @@ def generate_batch_norm_Variance():
194194

195195
def sample_predictor_configs(self, program_config):
196196
# for onednn
197-
if program_config.ops[0].attrs['use_mkldnn']:
197+
if program_config.ops[0].attrs['use_onednn']:
198198
config = self.create_inference_config(use_onednn=True)
199199
yield config, ['conv2d_transpose_bias'], (1e-5, 1e-5)
200200
# for cpu

test/ir/inference/test_conv_transpose_eltwiseadd_bn_fuse_pass.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -141,7 +141,7 @@ def generate_batch_norm_Variance():
141141
'data_format': random_data_layout,
142142
'output_size': random_output_size,
143143
'output_padding': random_output_size,
144-
'use_mkldnn': random_use_onednn,
144+
'use_onednn': random_use_onednn,
145145
'is_test': True,
146146
},
147147
)
@@ -182,7 +182,7 @@ def generate_batch_norm_Variance():
182182
'is_test': True,
183183
'trainable_statistics': False,
184184
'data_layout': random_data_layout,
185-
'use_mkldnn': random_use_onednn,
185+
'use_onednn': random_use_onednn,
186186
},
187187
)
188188

@@ -220,7 +220,7 @@ def generate_batch_norm_Variance():
220220

221221
def sample_predictor_configs(self, program_config):
222222
# for onednn
223-
if program_config.ops[2].attrs['use_mkldnn']:
223+
if program_config.ops[2].attrs['use_onednn']:
224224
config = self.create_inference_config(use_onednn=True)
225225
yield config, ['conv2d_transpose', 'elementwise_add'], (1e-5, 1e-5)
226226
# cpu

test/ir/inference/test_mkldnn_depthwise_conv_pass.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -99,7 +99,7 @@ def generate_conv2d_Filter():
9999
'paddings': random_paddings,
100100
'padding_algorithm': random_padding_algorithm,
101101
'data_format': random_data_layout,
102-
'use_mkldnn': True,
102+
'use_onednn': True,
103103
},
104104
)
105105

test/ir/inference/test_mkldnn_matmul_v2_activation_fuse_pass.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,7 @@ def generate_input(type):
8181
attrs={
8282
'trans_x': transpose_X,
8383
'trans_y': transpose_Y,
84-
'use_mkldnn': True,
84+
'use_onednn': True,
8585
},
8686
)
8787

0 commit comments

Comments
 (0)