support autocast bf16 for intel_hpu (#68445)

jychen21 · web-flow · commit 0fb6f12f7a15 · 2024-09-29T19:20:23.000+08:00
* support autocast bf16 for intel_hpu

* code style
diff --git a/python/paddle/amp/auto_cast.py b/python/paddle/amp/auto_cast.py
@@ -232,7 +232,10 @@ def _is_custom_device_bfloat16_supported() -> bool:
     Judge whether current custom device support bfloat16 amp.
     """
     place = _current_expected_place()
-    return place.get_device_type() == 'npu'
+    return (
+        place.get_device_type() == 'npu'
+        or place.get_device_type() == 'intel_hpu'
+    )
 
 
 def need_keep_fp32(layer: Layer, dtype: str) -> bool:
