未验证 提交 160dfd01 编写于 作者: Z Zhang Zheng 提交者: GitHub

Modify the condition of _get_places in fp16 (#52508)

上级 348a36b5
...@@ -2020,7 +2020,7 @@ class OpTest(unittest.TestCase): ...@@ -2020,7 +2020,7 @@ class OpTest(unittest.TestCase):
) )
def _get_places(self): def _get_places(self):
if self.dtype == np.float16: if self.dtype == np.float16 or self.dtype == "float16":
if core.is_compiled_with_cuda() and core.op_support_gpu( if core.is_compiled_with_cuda() and core.op_support_gpu(
self.op_type self.op_type
): ):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册