PaddlePaddle · Liyulingyue · Aug 10, 2023 · Aug 10, 2023 · Aug 13, 2023 · Aug 14, 2023
diff --git a/paddle/phi/api/yaml/static_ops.yaml b/paddle/phi/api/yaml/static_ops.yaml
@@ -90,7 +90,7 @@
   backward : assign_grad
 
 - op : assign_value
-  args : (int[] shape, DataType dtype, int[] bool_values = {}, float[] fp32_values = {}, int[] int32_values = {}, int64_t[] int64_values = {})
+  args : (int[] shape, DataType dtype, int[] bool_values = {}, float[] fp16_values = {}, float[] fp32_values = {}, int[] int32_values = {}, int64_t[] int64_values = {})
   output : Tensor(out)
   infer_meta :
     func : AssignValueInferMeta

diff --git a/paddle/phi/kernels/assign_kernel.cc b/paddle/phi/kernels/assign_kernel.cc
@@ -157,6 +157,7 @@ PD_REGISTER_KERNEL(assign_value,
                    ALL_LAYOUT,
                    phi::AssignValueKernel,
                    bool,
+                   phi::dtype::float16,
                    int,
                    float,
                    int8_t,

diff --git a/paddle/phi/ops/compat/assign_value_sig.cc b/paddle/phi/ops/compat/assign_value_sig.cc
@@ -36,6 +36,9 @@ KernelSignature AssignValueOpArgumentMapping(
   } else if (dtype == /*INT64*/ 3) {
     return KernelSignature(
         "assign_value", {}, {"shape", "dtype", "int64_values"}, {"Out"});
+  } else if (dtype == DataType::FLOAT16) {
+    return KernelSignature(
+        "assign_value", {}, {"shape", "dtype", "fp16_values"}, {"Out"});
   } else {
     return KernelSignature("unregistered", {}, {}, {});
   }
-Original file line number
+Diff line change
@@ Expand Up / @@ -157,6 +157,7 @@ PD_REGISTER_KERNEL(assign_value, @@
                        ALL_LAYOUT,
                        phi::AssignValueKernel,
                        bool,
+                       phi::dtype::float16,
                        int,
                        float,
                        int8_t,
@@ Expand Down @@