Skip to content

Commit 7487592

Browse files
authored
[BF16] Normal support bf16 (#54564) (#54620)
1 parent bd1a240 commit 7487592

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/paddle/tensor/random.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -327,7 +327,7 @@ def gaussian(shape, mean=0.0, std=1.0, seed=0, dtype=None, name=None):
327327
distribution, with ``shape`` and ``dtype``.
328328
"""
329329
op_type_for_check = 'gaussian/standard_normal/randn/normal'
330-
supported_dtypes = ['float32', 'float64', 'float16', 'uint16']
330+
supported_dtypes = ['float32', 'float64', 'float16', 'uint16', 'bfloat16']
331331

332332
if dtype is None:
333333
dtype = paddle.framework.get_default_dtype()

0 commit comments

Comments
 (0)