Skip to content

Commit 3c61529

Browse files
committed
fix a bug, create_parameter not supprot paddle.bfloat16
1 parent a7a2258 commit 3c61529

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

python/paddle/base/layer_helper_base.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,6 @@
2121
import paddle
2222

2323
from . import core, unique_name
24-
from .data_feeder import convert_dtype
2524
from .framework import (
2625
Variable,
2726
_current_expected_place,
@@ -368,7 +367,7 @@ def create_parameter(
368367
if not dtype:
369368
dtype = self.__dtype
370369
if isinstance(dtype, core.DataType):
371-
dtype = convert_dtype(dtype)
370+
dtype = paddle.pir.core.datatype_to_vartype[dtype]
372371
if is_bias:
373372
suffix = 'b'
374373
default_initializer = (

0 commit comments

Comments
 (0)