Skip to content

P
Paddle

PaddlePaddle / Paddle
8 个月 前同步成功

通知 2281
Star 20931
Fork 5422
P
Paddle

前往新版Gitcode，体验更适合开发者的 AI 搜索 >>

未验证 提交 e2b05dcc 编写于 作者: H hong 提交者： GitHub
浏览文件
操作

refine bilinear interp grad register (#56976)

上级 f5497fd0
隐藏空白更改
内联 并排
Showing with 5 addition and 0 deletion
paddle/phi/kernels/gpu/interpolate_grad_kernel.cu
浏览文件 @ e2b05dcc
......@@ -1572,6 +1572,7 @@ PD_REGISTER_KERNEL(bilinear_interp_grad,
double,
phi::dtype::float16,
phi::dtype::bfloat16) {
kernel->InputAt(1).SetBackend(phi::Backend::CPU);
kernel->InputAt(2).SetBackend(phi::Backend::ALL_BACKEND);
kernel->InputAt(3).SetBackend(phi::Backend::ALL_BACKEND);
}
......@@ -1583,6 +1584,7 @@ PD_REGISTER_KERNEL(nearest_interp_grad,
double,
phi::dtype::float16,
phi::dtype::bfloat16) {
kernel->InputAt(1).SetBackend(phi::Backend::CPU);
kernel->InputAt(2).SetBackend(phi::Backend::ALL_BACKEND);
kernel->InputAt(3).SetBackend(phi::Backend::ALL_BACKEND);
}
......@@ -1594,6 +1596,7 @@ PD_REGISTER_KERNEL(trilinear_interp_grad,
double,
phi::dtype::float16,
phi::dtype::bfloat16) {
kernel->InputAt(1).SetBackend(phi::Backend::CPU);
kernel->InputAt(2).SetBackend(phi::Backend::ALL_BACKEND);
kernel->InputAt(3).SetBackend(phi::Backend::ALL_BACKEND);
}
......@@ -1605,6 +1608,7 @@ PD_REGISTER_KERNEL(linear_interp_grad,
double,
phi::dtype::float16,
phi::dtype::bfloat16) {
kernel->InputAt(1).SetBackend(phi::Backend::CPU);
kernel->InputAt(2).SetBackend(phi::Backend::ALL_BACKEND);
kernel->InputAt(3).SetBackend(phi::Backend::ALL_BACKEND);
}
......@@ -1616,6 +1620,7 @@ PD_REGISTER_KERNEL(bicubic_interp_grad,
double,
phi::dtype::float16,
phi::dtype::bfloat16) {
kernel->InputAt(1).SetBackend(phi::Backend::CPU);
kernel->InputAt(2).SetBackend(phi::Backend::ALL_BACKEND);
kernel->InputAt(3).SetBackend(phi::Backend::ALL_BACKEND);
}
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或