From 74875921a68969bfe9c5d45638ee92c92a486c0e Mon Sep 17 00:00:00 2001 From: Zhong Hui Date: Wed, 14 Jun 2023 16:19:49 +0800 Subject: [PATCH] [BF16] Normal support bf16 (#54564) (#54620) --- python/paddle/tensor/random.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle/tensor/random.py b/python/paddle/tensor/random.py index 8683349c6e1..eb3ab19abc6 100644 --- a/python/paddle/tensor/random.py +++ b/python/paddle/tensor/random.py @@ -327,7 +327,7 @@ def gaussian(shape, mean=0.0, std=1.0, seed=0, dtype=None, name=None): distribution, with ``shape`` and ``dtype``. """ op_type_for_check = 'gaussian/standard_normal/randn/normal' - supported_dtypes = ['float32', 'float64', 'float16', 'uint16'] + supported_dtypes = ['float32', 'float64', 'float16', 'uint16', 'bfloat16'] if dtype is None: dtype = paddle.framework.get_default_dtype() -- GitLab