From 0adafe14d5c86b5ad0b9a3cae68fc1df6feacab8 Mon Sep 17 00:00:00 2001
From: huangjiyi <947613776@qq.com>
Date: Thu, 24 Apr 2025 17:06:29 +0800
Subject: [PATCH] [PHI] Register float16 & bfloat16 for assign_value kernel

---
 paddle/phi/kernels/assign_kernel.cc | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/paddle/phi/kernels/assign_kernel.cc b/paddle/phi/kernels/assign_kernel.cc
index abba15cfbaffa..d0d03f863b902 100644
--- a/paddle/phi/kernels/assign_kernel.cc
+++ b/paddle/phi/kernels/assign_kernel.cc
@@ -138,6 +138,8 @@ PD_REGISTER_KERNEL(assign_value,
                    double,
                    int8_t,
                    int64_t,
+                   phi::dtype::float16,
+                   phi::dtype::bfloat16,
                    phi::dtype::complex<float>,
                    phi::dtype::complex<double>) {}
 
@@ -168,6 +170,8 @@ PD_REGISTER_KERNEL(assign_value,
                    double,
                    int8_t,
                    int64_t,
+                   phi::dtype::float16,
+                   phi::dtype::bfloat16,
                    phi::dtype::complex<float>,
                    phi::dtype::complex<double>) {}
 #endif