add data op data type (#56033)

7472057c · hong · GitHub · 611e71d0 · 7472057c · 7472057c
隐藏空白更改
内联并排

Showing with 6 addition and 0 deletion

paddle/phi/kernels/cpu/data_kernel.cc paddle/phi/kernels/cpu/data_kernel.cc +3 -0

paddle/phi/kernels/gpu/data_kernel.cu paddle/phi/kernels/gpu/data_kernel.cu +3 -0

未找到文件。
--- a/paddle/phi/kernels/cpu/data_kernel.cc
+++ b/paddle/phi/kernels/cpu/data_kernel.cc
@@ -42,7 +42,10 @@ PD_REGISTER_KERNEL(shadow_feed,
                   ALL_LAYOUT,
                   phi::ShadowFeedKernel,
                   bool,
+                   uint8_t,
                   float,
+                   int8_t,
+                   int16_t,
                   int32_t,
                   int64_t,
                   double,

--- a/paddle/phi/kernels/gpu/data_kernel.cu
+++ b/paddle/phi/kernels/gpu/data_kernel.cu
@@ -23,7 +23,10 @@ PD_REGISTER_KERNEL(shadow_feed,
                   ALL_LAYOUT,
                   phi::ShadowFeedKernel,
                   bool,
+                   uint8_t,
                   float,
+                   int8_t,
+                   int16_t,
                   int32_t,
                   int64_t,
                   double,