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 9bf1ff43a26e96d5463faabad8818f1691b30346..4772ae00d3fc36a30b4e55cd73398f4d97b51028 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