fix bug in amp-bf16 (#60268)

Author: heavyrain-lzy

python/paddle/distributed/auto_parallel/static/engine.py

@@ -848,7 +848,9 @@ def _initialize(self, mode, init_parameters=True):
  # for amp
  if dest_type == core.VarDesc.VarType.BF16:
  buffer_tensor.set(
- _convert_float_to_bfloat16(buffer.numpy()),
+ _convert_float_to_bfloat16(
+ self._place, buffer.numpy()
+ ),
  self._place,
  )
  elif dest_type == core.VarDesc.VarType.FP16: