未验证 提交 7b1f42d3 编写于 作者: D duanyanhui 提交者: GitHub

add custom_cpu mixed_precision test (#50789)

上级 df7cc3a0
...@@ -57,6 +57,7 @@ class TestCustomCPUPlugin(unittest.TestCase): ...@@ -57,6 +57,7 @@ class TestCustomCPUPlugin(unittest.TestCase):
self._test_fallback_kernel() self._test_fallback_kernel()
self._test_scalar() self._test_scalar()
self._test_custom_device_py_api() self._test_custom_device_py_api()
self._test_custom_device_mix_precision()
def _test_custom_device_dataloader(self): def _test_custom_device_dataloader(self):
import paddle import paddle
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册