From a7d306afe64c29e8df5cb613cf197d123f312d8b Mon Sep 17 00:00:00 2001
From: Yuang Liu <liuyuang@baidu.com>
Date: Fri, 18 Nov 2022 12:09:35 +0800
Subject: [PATCH] add bf16 for numel (#48121)

---
 paddle/phi/kernels/cpu/size_kernel.cc | 1 +
 paddle/phi/kernels/gpu/size_kernel.cu | 1 +
 2 files changed, 2 insertions(+)

diff --git a/paddle/phi/kernels/cpu/size_kernel.cc b/paddle/phi/kernels/cpu/size_kernel.cc
index 4019976ecec..a070c2de5bc 100644
--- a/paddle/phi/kernels/cpu/size_kernel.cc
+++ b/paddle/phi/kernels/cpu/size_kernel.cc
@@ -27,6 +27,7 @@ PD_REGISTER_KERNEL(size,
                    int,
                    int64_t,
                    phi::dtype::float16,
+                   phi::dtype::bfloat16,
                    float,
                    double,
                    bool) {}
diff --git a/paddle/phi/kernels/gpu/size_kernel.cu b/paddle/phi/kernels/gpu/size_kernel.cu
index fb6acd5599a..a165e6c82df 100644
--- a/paddle/phi/kernels/gpu/size_kernel.cu
+++ b/paddle/phi/kernels/gpu/size_kernel.cu
@@ -26,6 +26,7 @@ PD_REGISTER_KERNEL(size,
                    int,
                    int64_t,
                    phi::dtype::float16,
+                   phi::dtype::bfloat16,
                    float,
                    double,
                    bool) {}
-- 
GitLab