未验证 提交 a90711c7 编写于 作者: J joanna.wozna.intel 提交者: GitHub

Add avx512 core instructions check 2 (#27750)

* Add test skip from cmake

* Remove print
上级 f399bed8
...@@ -35,6 +35,8 @@ def conv2d_residual_naive(out, residual): ...@@ -35,6 +35,8 @@ def conv2d_residual_naive(out, residual):
return out return out
@unittest.skipIf(not core.supports_bfloat16(),
"place does not support BF16 evaluation")
class TestConv2dBf16Op(TestConv2dOp): class TestConv2dBf16Op(TestConv2dOp):
def setUp(self): def setUp(self):
self.op_type = "conv2d" self.op_type = "conv2d"
...@@ -42,9 +44,9 @@ class TestConv2dBf16Op(TestConv2dOp): ...@@ -42,9 +44,9 @@ class TestConv2dBf16Op(TestConv2dOp):
self.exhaustive_search = False self.exhaustive_search = False
self.use_cuda = False self.use_cuda = False
self.use_mkldnn = True self.use_mkldnn = True
self._cpu_only = True
self.weight_type = np.float32 self.weight_type = np.float32
self.input_type = np.float32 self.input_type = np.float32
self.use_mkldnn = True
self.mkldnn_data_type = "bfloat16" self.mkldnn_data_type = "bfloat16"
self.force_fp32_output = False self.force_fp32_output = False
self.init_group() self.init_group()
...@@ -205,5 +207,4 @@ class TestWithInput1x1Filter1x1(TestConv2dBf16Op): ...@@ -205,5 +207,4 @@ class TestWithInput1x1Filter1x1(TestConv2dBf16Op):
if __name__ == '__main__': if __name__ == '__main__':
if core.supports_bfloat16(): unittest.main()
unittest.main()
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册