Skip to content

Commit f9b74fc

Browse files
authored
rename mkldnn to onednn in test/onednn/ (#75603)
1 parent a5eff51 commit f9b74fc

File tree

4 files changed

+8
-8
lines changed

4 files changed

+8
-8
lines changed

test/onednn/test_flags_onednn_ops_on_off.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222
class TestFlagsUseOnednn(unittest.TestCase):
2323
def setUp(self):
2424
self._python_interp = sys.executable
25-
self._python_interp += " check_flags_mkldnn_ops_on_off.py"
25+
self._python_interp += " check_flags_onednn_ops_on_off.py"
2626

2727
self.env = os.environ.copy()
2828
self.env["DNNL_VERBOSE"] = "1"

test/onednn/test_matmul_bf16_onednn_op.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ def set_attributes(self):
3434
self.attrs = {
3535
'alpha': self.alpha,
3636
"use_onednn": self.use_onednn,
37-
"mkldnn_data_type": self.onednn_data_type,
37+
"onednn_data_type": self.onednn_data_type,
3838
"force_fp32_output": self.force_fp32_output,
3939
'transpose_X': False,
4040
'transpose_Y': False,
@@ -147,7 +147,7 @@ def generate_data(self):
147147
def set_attributes(self):
148148
self.attrs = {
149149
"use_onednn": self.use_onednn,
150-
"mkldnn_data_type": self.onednn_data_type,
150+
"onednn_data_type": self.onednn_data_type,
151151
'transpose_X': True,
152152
'transpose_Y': False,
153153
}
@@ -162,7 +162,7 @@ def generate_data(self):
162162
def set_attributes(self):
163163
self.attrs = {
164164
"use_onednn": self.use_onednn,
165-
"mkldnn_data_type": self.onednn_data_type,
165+
"onednn_data_type": self.onednn_data_type,
166166
'transpose_Y': True,
167167
'transpose_X': False,
168168
}

test/onednn/test_matmul_v2_onednn_op.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ def set_inputs(self, x, y):
6060
self.inputs = {'X': x, 'Y': y}
6161

6262
def set_dtype_attr(self):
63-
self.attrs['mkldnn_data_type'] = "float32"
63+
self.attrs['onednn_data_type'] = "float32"
6464

6565
def setUp(self):
6666
self.config()
@@ -313,7 +313,7 @@ def set_inputs(self, x, y):
313313
self.y_fp32 = y
314314

315315
def set_dtype_attr(self):
316-
self.attrs['mkldnn_data_type'] = "bfloat16"
316+
self.attrs['onednn_data_type'] = "bfloat16"
317317

318318
def test_check_output(self):
319319
self.check_output_with_place(

test/onednn/test_pool2d_int8_onednn_op.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -123,8 +123,8 @@ class TestU8Case(parent):
123123
def init_data_type(self):
124124
self.dtype = np.uint8
125125

126-
cls_name_s8 = "{}_{}".format(parent.__name__, "mkldnn_s8")
127-
cls_name_u8 = "{}_{}".format(parent.__name__, "mkldnn_u8")
126+
cls_name_s8 = "{}_{}".format(parent.__name__, "onednn_s8")
127+
cls_name_u8 = "{}_{}".format(parent.__name__, "onednn_u8")
128128
TestS8Case.__name__ = cls_name_s8
129129
TestU8Case.__name__ = cls_name_u8
130130
globals()[cls_name_s8] = TestS8Case

0 commit comments

Comments
 (0)