Rename use_mkldnn to use_onednn in test cases - part6

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__':