Rename use_mkldnn to use_onednn in test cases - part7

test/mkldnn/test_elementwise_sub_onednn_op.py

@@ -44,7 +44,7 @@ def setUp(self):
  'X': OpTest.np_dtype_to_base_dtype(self.x),
  'Y': OpTest.np_dtype_to_base_dtype(self.y),
  }
- self.attrs = {'axis': self.axis, 'use_mkldnn': self.use_mkldnn}
+ self.attrs = {'axis': self.axis, 'use_mkldnn': self.use_onednn}
  self.outputs = {'Out': self.out}
 
  def init_input_output(self):
@@ -71,7 +71,7 @@ def init_axis(self):
  self.axis = -1
 
  def init_kernel_type(self):
- self.use_mkldnn = True
+ self.use_onednn = True
 
  def init_dtype(self):
  self.dtype = np.float32
@@ -225,12 +225,12 @@ def setUp(self):
  self.x_bf16 = convert_float_to_uint16(self.x)
  self.y_bf16 = convert_float_to_uint16(self.y)
  self.inputs = {'X': self.x_bf16, 'Y': self.y_bf16}
- self.attrs = {'axis': self.axis, 'use_mkldnn': self.use_mkldnn}
+ self.attrs = {'axis': self.axis, 'use_mkldnn': self.use_onednn}
  self.outputs = {'Out': convert_float_to_uint16(self.out)}
 
  def init_dtype(self):
  self.dtype = np.float32
- self.mkldnn_data_type = "bfloat16"
+ self.onednn_data_type = "bfloat16"
 
  def init_input_output(self):
  self.x = np.random.random(
@@ -312,7 +312,7 @@ def test_check_grad_ignore_x(self):
 # complex64, int16, float64, bfloat16, complex128, float32, int32, int64
 '''class TestInt8(TestOneDNNElementwiseSubOp):
  def init_kernel_type(self):
- self.use_mkldnn = True
+ self.use_onednn = True
  self._cpu_only = True
 
  def init_dtype(self):

test/mkldnn/test_fc_bf16_mkldnn_op.py

@@ -42,8 +42,8 @@ def generate_data(self):
 
  def setUp(self):
  self.op_type = "fc"
- self.use_mkldnn = True
- self.mkldnn_data_type = "bfloat16"
+ self.use_onednn = True
+ self.onednn_data_type = "bfloat16"
  self.force_fp32_output = False
  self.generate_data()
 
@@ -60,7 +60,7 @@ def setUp(self):
  }
 
  self.attrs = {
- 'use_mkldnn': self.use_mkldnn,
+ 'use_mkldnn': self.use_onednn,
  'force_fp32_output': self.force_fp32_output,
  }
 

test/mkldnn/test_fc_mkldnn_op.py

@@ -37,15 +37,15 @@ def create_data(self):
  def setUp(self):
  self.op_type = "fc"
  self._cpu_only = True
- self.use_mkldnn = True
+ self.use_onednn = True
  self.create_data()
  self.inputs = {
  'Input': self.matrix.input,
  'W': self.matrix.weights,
  'Bias': self.bias,
  }
 
- self.attrs = {'use_mkldnn': self.use_mkldnn}
+ self.attrs = {'use_mkldnn': self.use_onednn}
 
  self.outputs = {
  'Out': fully_connected_naive(