Fix

Author: co63oc

test/ir/inference/test_mkldnn_conv3d_op.py

@@ -123,7 +123,7 @@ def generate_weight(*args, **kwargs):
                 "paddings": kwargs["paddings"],
                 "strides": kwargs["strides"],
                 "is_test": True,
-                "use_mkldnn": True,
+                "use_onednn": True,
             },
         )
 

test/ir/inference/test_mkldnn_matmul_elementwise_add_activation_fuse_pass.py

@@ -60,7 +60,7 @@ def generate_input():
             inputs={'X': ['matmul_x'], 'Y': ['matmul_y']},
             outputs={'Out': ['matmul_output']},
             attrs={
-                'use_mkldnn': True,
+                'use_onednn': True,
             },
         )
 
@@ -73,7 +73,7 @@ def generate_input():
             type='elementwise_add',
             inputs=inputs,
             outputs={'Out': ['elementwise_add_output']},
-            attrs={'axis': axis, 'use_mkldnn': True},
+            attrs={'axis': axis, 'use_onednn': True},
         )
 
         if activation_type == "relu6":

test/ir/inference/test_onednn_fc_activation_fuse_pass.py

@@ -56,7 +56,7 @@ def generate_input(shape):
             },
             outputs={"Out": ["fc_output"]},
             attrs={
-                "use_mkldnn": True,
+                "use_onednn": True,
                 "padding_weights": False,
                 "in_num_col_dims": 1,
             },