Skip to content

Commit 6dc966a

Browse files
authored
Rename use_mkldnn to use_onednn in test cases - part2 (#73902)
1 parent 70f2458 commit 6dc966a

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

test/mkldnn/test_sum_bf16_mkldnn_op.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -28,8 +28,8 @@
2828
class TestSumBF16ONEDNN(TestSumOp):
2929
def setUp(self):
3030
self.op_type = "sum"
31-
self.use_mkldnn = True
32-
self.mkldnn_data_type = "bfloat16"
31+
self.use_onednn = True
32+
self.onednn_data_type = "bfloat16"
3333

3434
# float32 input to be use for reference
3535
x0 = np.random.random((25, 8)).astype('float32')
@@ -45,7 +45,7 @@ def setUp(self):
4545

4646
y = x0 + x1 + x2
4747
self.outputs = {'Out': convert_float_to_uint16(y)}
48-
self.attrs = {'use_mkldnn': self.use_mkldnn}
48+
self.attrs = {'use_mkldnn': self.use_onednn}
4949

5050
def test_check_output(self):
5151
self.check_output_with_place(core.CPUPlace(), check_pir_onednn=True)

0 commit comments

Comments
 (0)