Fix

Author: co63oc

test/ir/inference/test_conv_transpose_bn_fuse_pass.py

@@ -133,7 +133,7 @@ def generate_batch_norm_Variance():
                 'data_format': random_data_layout,
                 'output_size': random_output_size,
                 'output_padding': random_output_size,
-                'use_mkldnn': random_use_onednn,
+                'use_onednn': random_use_onednn,
                 'is_test': True,
             },
         )
@@ -160,7 +160,7 @@ def generate_batch_norm_Variance():
                 'is_test': True,
                 'trainable_statistics': False,
                 'data_layout': random_data_layout,
-                'use_mkldnn': random_use_onednn,
+                'use_onednn': random_use_onednn,
             },
         )
 
@@ -194,7 +194,7 @@ def generate_batch_norm_Variance():
 
     def sample_predictor_configs(self, program_config):
         # for onednn
-        if program_config.ops[0].attrs['use_mkldnn']:
+        if program_config.ops[0].attrs['use_onednn']:
             config = self.create_inference_config(use_onednn=True)
             yield config, ['conv2d_transpose_bias'], (1e-5, 1e-5)
         # for cpu

test/ir/inference/test_conv_transpose_eltwiseadd_bn_fuse_pass.py

@@ -141,7 +141,7 @@ def generate_batch_norm_Variance():
                 'data_format': random_data_layout,
                 'output_size': random_output_size,
                 'output_padding': random_output_size,
-                'use_mkldnn': random_use_onednn,
+                'use_onednn': random_use_onednn,
                 'is_test': True,
             },
         )
@@ -182,7 +182,7 @@ def generate_batch_norm_Variance():
                 'is_test': True,
                 'trainable_statistics': False,
                 'data_layout': random_data_layout,
-                'use_mkldnn': random_use_onednn,
+                'use_onednn': random_use_onednn,
             },
         )
 
@@ -220,7 +220,7 @@ def generate_batch_norm_Variance():
 
     def sample_predictor_configs(self, program_config):
         # for onednn
-        if program_config.ops[2].attrs['use_mkldnn']:
+        if program_config.ops[2].attrs['use_onednn']:
             config = self.create_inference_config(use_onednn=True)
             yield config, ['conv2d_transpose', 'elementwise_add'], (1e-5, 1e-5)
         # cpu

test/ir/inference/test_mkldnn_depthwise_conv_pass.py

@@ -99,7 +99,7 @@ def generate_conv2d_Filter():
                 'paddings': random_paddings,
                 'padding_algorithm': random_padding_algorithm,
                 'data_format': random_data_layout,
-                'use_mkldnn': True,
+                'use_onednn': True,
             },
         )
 

test/ir/inference/test_mkldnn_matmul_activation_fuse_pass.py

@@ -77,7 +77,7 @@ def generate_input(type):
                 'transpose_X': transpose_X,
                 'transpose_Y': transpose_Y,
                 'alpha': alpha,
-                'use_mkldnn': True,
+                'use_onednn': True,
             },
         )
 

test/ir/inference/test_mkldnn_matmul_v2_activation_fuse_pass.py

@@ -81,7 +81,7 @@ def generate_input(type):
             attrs={
                 'trans_x': transpose_X,
                 'trans_y': transpose_Y,
-                'use_mkldnn': True,
+                'use_onednn': True,
             },
         )
 

test/ir/inference/test_onednn_operator_reshape2_fuse_pass.py

@@ -45,7 +45,7 @@ def generate_input(shape):
             },
             attrs={
                 "axis": axis,
-                "use_mkldnn": True,
+                "use_onednn": True,
             },
         )
 

test/ir/inference/test_onednn_squeeze2_transpose2_fuse_pass.py

@@ -42,7 +42,7 @@ def generate_input(shape):
             },
             attrs={
                 "axes": [2],
-                "use_mkldnn": True,
+                "use_onednn": True,
             },
         )
 
@@ -57,7 +57,7 @@ def generate_input(shape):
             },
             attrs={
                 "axis": transpose_axis,
-                "use_mkldnn": True,
+                "use_onednn": True,
             },
         )