Fix

Author: co63oc

test/mkldnn/test_fusion_gru_int8_mkldnn_op.py

@@ -35,8 +35,8 @@ def setUp(self):
         self.act_state = 'tanh'
         self.act_gate = 'sigmoid'
         self.origin_mode = True
-        self.use_mkldnn = True
-        self.mkldnn_data_type = "int8"
+        self.use_onednn = True
+        self.onednn_data_type = "int8"
         self.force_fp32_output = True
         self.error_margin = 1e-5
         self.set_confs()
@@ -141,8 +141,8 @@ def setUp(self):
             'gate_activation': self.act_gate,
             'is_reverse': self.is_reverse,
             'origin_mode': self.origin_mode,
-            'use_mkldnn': self.use_mkldnn,
-            'mkldnn_data_type': self.mkldnn_data_type,
+            'use_mkldnn': self.use_onednn,
+            'mkldnn_data_type': self.onednn_data_type,
             'force_fp32_output': self.force_fp32_output,
             'Scale_data': scale_data,
             'Shift_data': shift_data,

test/mkldnn/test_fusion_lstm_bf16_mkldnn_op.py

@@ -50,8 +50,8 @@ def setUp(self):
         self.act_gate = 'sigmoid'
         self.act_cell = 'tanh'
         self.act_cand = 'tanh'
-        self.use_mkldnn = True
-        self.mkldnn_data_type = "bfloat16"
+        self.use_onednn = True
+        self.onednn_data_type = "bfloat16"
         self.force_fp32_output = False
         self.weights_dtype = 'fp32'
         self.set_confs()
@@ -145,8 +145,8 @@ def setUp(self):
             'cell_activation': self.act_cell,
             'candidate_activation': self.act_cand,
             'force_fp32_output': self.force_fp32_output,
-            'use_mkldnn': self.use_mkldnn,
-            'mkldnn_data_type': self.mkldnn_data_type,
+            'use_mkldnn': self.use_onednn,
+            'mkldnn_data_type': self.onednn_data_type,
         }
 
 

test/mkldnn/test_fusion_lstm_int8_mkldnn_op.py

@@ -34,8 +34,8 @@ def setUp(self):
         self.act_gate = 'sigmoid'
         self.act_cand = 'tanh'
         self.use_peepholes = False  # LSTM u8 doesn't support peepholes
-        self.use_mkldnn = True
-        self.mkldnn_data_type = "int8"
+        self.use_onednn = True
+        self.onednn_data_type = "int8"
         self.force_fp32_output = False
         self.error_margin = 1e-5
         self.set_confs()
@@ -130,8 +130,8 @@ def setUp(self):
             'candidate_activation': self.act_cand,
             'is_reverse': self.is_reverse,
             'use_peepholes': self.use_peepholes,
-            'use_mkldnn': self.use_mkldnn,
-            'mkldnn_data_type': self.mkldnn_data_type,
+            'use_mkldnn': self.use_onednn,
+            'mkldnn_data_type': self.onednn_data_type,
             'force_fp32_output': self.force_fp32_output,
             'Scale_data': scale_data,
             'Shift_data': shift_data,

test/mkldnn/test_fusion_lstm_mkldnn_op.py

@@ -19,7 +19,7 @@
 
 class TestFusionLSTMONEDNNOp(TestFusionLSTMOp):
     def set_conf(self):
-        self.use_mkldnn = True
+        self.use_onednn = True
         self.check_pir_onednn = True
 
     def test_check_output(self):
@@ -35,49 +35,49 @@ def test_check_output(self):
 class TestFusionLSTMONEDNNOpReverse(TestFusionLSTMONEDNNOp):
     def set_conf(self):
         self.is_reverse = True
-        self.use_mkldnn = True
+        self.use_onednn = True
 
 
 class TestFusionLSTMONEDNNOpInitReverse(TestFusionLSTMONEDNNOp):
     def set_conf(self):
         self.has_initial_state = True
         self.is_reverse = True
-        self.use_mkldnn = True
+        self.use_onednn = True
 
 
 class TestFusionLSTMONEDNNOpMD1(TestFusionLSTMONEDNNOp):
     def set_conf(self):
         self.M = 36
         self.D = 8
-        self.use_mkldnn = True
+        self.use_onednn = True
 
 
 class TestFusionLSTMONEDNNOpMD2(TestFusionLSTMONEDNNOp):
     def set_conf(self):
         self.M = 8
         self.D = 8
-        self.use_mkldnn = True
+        self.use_onednn = True
 
 
 class TestFusionLSTMONEDNNOpMD3(TestFusionLSTMONEDNNOp):
     def set_conf(self):
         self.M = 15
         self.D = 3
-        self.use_mkldnn = True
+        self.use_onednn = True
 
 
 class TestFusionLSTMONEDNNOpBS1(TestFusionLSTMONEDNNOp):
     def set_conf(self):
         self.lod = [[3]]
         self.D = 16
-        self.use_mkldnn = True
+        self.use_onednn = True
 
 
 class TestFusionLSTMONEDNNOpPeepholesInit(TestFusionLSTMONEDNNOp):
     def set_conf(self):
         self.use_peepholes = True
         self.has_initial_state = True
-        self.use_mkldnn = True
+        self.use_onednn = True
 
 
 if __name__ == '__main__':