fix a bug, create_parameter not supprot paddle.bfloat16 (#72451)

wanghuancoder · web-flow · commit e21615a0fbd8 · 2025-04-25T09:56:40.000+08:00
* fix a bug, create_parameter not supprot paddle.bfloat16
diff --git a/python/paddle/base/layer_helper_base.py b/python/paddle/base/layer_helper_base.py
@@ -21,7 +21,6 @@
 import paddle
 
 from . import core, unique_name
-from .data_feeder import convert_dtype
 from .framework import (
     Variable,
     _current_expected_place,
@@ -368,7 +367,7 @@ def create_parameter(
         if not dtype:
             dtype = self.__dtype
         if isinstance(dtype, core.DataType):
-            dtype = convert_dtype(dtype)
+            dtype = paddle.pir.core.datatype_to_vartype[dtype]
         if is_bias:
             suffix = 'b'
             default_initializer = (
@@ -449,6 +448,8 @@ def create_parameter(
             )
         else:
             if in_pir_mode():
+                if isinstance(dtype, core.VarDesc.VarType):
+                    dtype = paddle.pir.core.vartype_to_datatype[dtype]
                 return paddle.pir.core.create_parameter(
                     dtype=dtype,
                     shape=shape,
