Skip to content

P
Paddle

机器未来 / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
未验证 提交 b051455f 编写于 作者: F Feiyu Chan 提交者： GitHub
浏览文件
操作

set_value_op: add support for complex types (#46885)

上级 2190da20
隐藏空白更改
内联 并排
Showing with 22 addition and 6 deletion
paddle/phi/kernels/cpu/set_value_grad_kernel.cc
浏览文件 @ b051455f
......@@ -15,6 +15,7 @@
#include "paddle/phi/kernels/set_value_grad_kernel.h"
#include "paddle/phi/backends/cpu/cpu_context.h"
#include "paddle/phi/common/complex.h"
#include "paddle/phi/core/kernel_registry.h"
#include "paddle/phi/kernels/impl/set_value_grad_kernel_impl.h"
......@@ -27,4 +28,6 @@ PD_REGISTER_KERNEL(set_value_grad,
int,
int64_t,
bool,
phi::dtype::float16) {}
phi::dtype::float16,
phi::dtype::complex<float>,
phi::dtype::complex<double>) {}
paddle/phi/kernels/cpu/set_value_kernel.cc
浏览文件 @ b051455f
......@@ -15,6 +15,7 @@
#include "paddle/phi/kernels/set_value_kernel.h"
#include "paddle/phi/backends/cpu/cpu_context.h"
#include "paddle/phi/common/complex.h"
#include "paddle/phi/core/kernel_registry.h"
#include "paddle/phi/kernels/impl/set_value_kernel_impl.h"
......@@ -27,7 +28,9 @@ PD_REGISTER_KERNEL(set_value,
int,
int64_t,
bool,
phi::dtype::float16) {}
phi::dtype::float16,
phi::dtype::complex<float>,
phi::dtype::complex<double>) {}
PD_REGISTER_KERNEL(set_value_with_tensor,
CPU,
ALL_LAYOUT,
......@@ -37,4 +40,6 @@ PD_REGISTER_KERNEL(set_value_with_tensor,
int,
int64_t,
bool,
phi::dtype::float16) {}
phi::dtype::float16,
phi::dtype::complex<float>,
phi::dtype::complex<double>) {}
paddle/phi/kernels/gpu/set_value_grad_kernel.cu
浏览文件 @ b051455f
......@@ -15,6 +15,7 @@
#include "paddle/phi/kernels/set_value_grad_kernel.h"
#include "paddle/phi/backends/gpu/gpu_context.h"
#include "paddle/phi/common/complex.h"
#include "paddle/phi/core/kernel_registry.h"
#include "paddle/phi/kernels/impl/set_value_grad_kernel_impl.h"
......@@ -27,4 +28,6 @@ PD_REGISTER_KERNEL(set_value_grad,
int,
int64_t,
bool,
phi::dtype::float16) {}
phi::dtype::float16,
phi::dtype::complex<float>,
phi::dtype::complex<double>) {}
paddle/phi/kernels/gpu/set_value_kernel.cu
浏览文件 @ b051455f
......@@ -15,6 +15,7 @@
#include "paddle/phi/kernels/set_value_kernel.h"
#include "paddle/phi/backends/gpu/gpu_context.h"
#include "paddle/phi/common/complex.h"
#include "paddle/phi/core/kernel_registry.h"
#include "paddle/phi/kernels/impl/set_value_kernel_impl.h"
......@@ -27,7 +28,9 @@ PD_REGISTER_KERNEL(set_value,
int,
int64_t,
bool,
paddle::platform::float16) {}
paddle::platform::float16,
phi::dtype::complex<float>,
phi::dtype::complex<double>) {}
PD_REGISTER_KERNEL(set_value_with_tensor,
GPU,
ALL_LAYOUT,
......@@ -37,4 +40,6 @@ PD_REGISTER_KERNEL(set_value_with_tensor,
int,
int64_t,
bool,
paddle::platform::float16) {}
paddle::platform::float16,
phi::dtype::complex<float>,
phi::dtype::complex<double>) {}
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或