未验证 提交 c828934e 编写于 作者: J jiangcheng 提交者: GitHub

[CINN] fix check_cinn in Optest check condition error bug (#53676)

上级 f023d42f
...@@ -557,9 +557,9 @@ class OpTest(unittest.TestCase): ...@@ -557,9 +557,9 @@ class OpTest(unittest.TestCase):
def _enable_check_cinn_test(self, place, inputs, outputs): def _enable_check_cinn_test(self, place, inputs, outputs):
# if the test not run in cuda or the paddle not compile with CINN, skip cinn test # if the test not run in cuda or the paddle not compile with CINN, skip cinn test
if ( if (
core.is_compiled_with_cinn() not core.is_compiled_with_cinn()
and core.is_compiled_with_cuda() or not core.is_compiled_with_cuda()
and isinstance(place, fluid.CUDAPlace) or not isinstance(place, fluid.CUDAPlace)
): ):
return False return False
# CINN not support bfloat16 now, skip cinn test # CINN not support bfloat16 now, skip cinn test
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册