From aa60a7b19811693374b3f956b7058713b4484dc1 Mon Sep 17 00:00:00 2001
From: Leo Chen <chenqiuliang@baidu.com>
Date: Wed, 6 Sep 2023 10:39:12 +0800
Subject: [PATCH] add bfloat16 in support lists (#56989)

---
 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 109dbcdfd97..292d37a1015 100644
--- a/python/paddle/tensor/random.py
+++ b/python/paddle/tensor/random.py
@@ -324,7 +324,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