adjust unittest. (#52847)

b9ccf0e6 · liuruyan · GitHub · 5a864270 · b9ccf0e6
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

test/amp/test_layer_convert_dtype.py test/amp/test_layer_convert_dtype.py +1 -1

未找到文件。
--- a/test/amp/test_layer_convert_dtype.py
+++ b/test/amp/test_layer_convert_dtype.py
@@ -127,7 +127,7 @@ class TestSupportedTypeInfo(unittest.TestCase):
        res = paddle.amp.is_float16_supported('cpu')
        self.assertEqual(res, False)
        res = paddle.amp.is_bfloat16_supported('cpu')
-        self.assertEqual(res, True)
+        self.assertEqual(res, core.supports_bfloat16())

    def test_gpu_fp16_supported(self):
        res = paddle.amp.is_float16_supported()