diff --git a/python/paddle/fluid/tests/unittests/eager_op_test.py b/python/paddle/fluid/tests/unittests/eager_op_test.py index 362ade417257c430a0b3576d9790b0698409f5df..ae75ac02f9450ef1dd50edb6062ee10d1a8fe4a9 100644 --- a/python/paddle/fluid/tests/unittests/eager_op_test.py +++ b/python/paddle/fluid/tests/unittests/eager_op_test.py @@ -2020,7 +2020,7 @@ class OpTest(unittest.TestCase): ) 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( self.op_type ):