There was an error while loading. Please reload this page.
1 parent 70f2458 commit 6dc966aCopy full SHA for 6dc966a
test/mkldnn/test_sum_bf16_mkldnn_op.py
@@ -28,8 +28,8 @@
28
class TestSumBF16ONEDNN(TestSumOp):
29
def setUp(self):
30
self.op_type = "sum"
31
- self.use_mkldnn = True
32
- self.mkldnn_data_type = "bfloat16"
+ self.use_onednn = True
+ self.onednn_data_type = "bfloat16"
33
34
# float32 input to be use for reference
35
x0 = np.random.random((25, 8)).astype('float32')
@@ -45,7 +45,7 @@ def setUp(self):
45
46
y = x0 + x1 + x2
47
self.outputs = {'Out': convert_float_to_uint16(y)}
48
- self.attrs = {'use_mkldnn': self.use_mkldnn}
+ self.attrs = {'use_mkldnn': self.use_onednn}
49
50
def test_check_output(self):
51
self.check_output_with_place(core.CPUPlace(), check_pir_onednn=True)
0 commit comments