From 7b1f42d3883dc0eab7e08346cad63256027c51fe Mon Sep 17 00:00:00 2001 From: duanyanhui <45005871+YanhuiDua@users.noreply.github.com> Date: Thu, 23 Feb 2023 15:52:33 +0800 Subject: [PATCH] add custom_cpu mixed_precision test (#50789) --- .../paddle/fluid/tests/custom_runtime/test_custom_cpu_plugin.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/paddle/fluid/tests/custom_runtime/test_custom_cpu_plugin.py b/python/paddle/fluid/tests/custom_runtime/test_custom_cpu_plugin.py index 9bf1ff43a26..4772ae00d3f 100755 --- a/python/paddle/fluid/tests/custom_runtime/test_custom_cpu_plugin.py +++ b/python/paddle/fluid/tests/custom_runtime/test_custom_cpu_plugin.py @@ -57,6 +57,7 @@ class TestCustomCPUPlugin(unittest.TestCase): self._test_fallback_kernel() self._test_scalar() self._test_custom_device_py_api() + self._test_custom_device_mix_precision() def _test_custom_device_dataloader(self): import paddle -- GitLab