Skip to content

Commit

Permalink
fix set backend (PaddlePaddle#63397)
Browse files Browse the repository at this point in the history
  • Loading branch information
gouzil authored Apr 10, 2024
1 parent b7714e2 commit 4eb91a6
Show file tree
Hide file tree
Showing 4 changed files with 0 additions and 6 deletions.
1 change: 0 additions & 1 deletion paddle/phi/kernels/cpu/full_kernel.cc
Original file line number Diff line number Diff line change
Expand Up @@ -156,5 +156,4 @@ PD_REGISTER_KERNEL(full_with_tensor,
phi::dtype::complex<float>,
phi::dtype::complex<double>) {
kernel->InputAt(0).SetBackend(phi::Backend::CPU);
kernel->InputAt(1).SetBackend(phi::Backend::CPU);
}
1 change: 0 additions & 1 deletion paddle/phi/kernels/gpu/full_kernel.cu
Original file line number Diff line number Diff line change
Expand Up @@ -174,5 +174,4 @@ PD_REGISTER_KERNEL(full_with_tensor,
phi::dtype::complex<float>,
phi::dtype::complex<double>) {
kernel->InputAt(0).SetBackend(phi::Backend::CPU);
kernel->InputAt(1).SetBackend(phi::Backend::CPU);
}
3 changes: 0 additions & 3 deletions paddle/phi/kernels/selected_rows/full_kernel.cc
Original file line number Diff line number Diff line change
Expand Up @@ -111,7 +111,6 @@ PD_REGISTER_KERNEL(full_with_tensor_sr,
phi::dtype::complex<float>,
phi::dtype::complex<double>) {
kernel->InputAt(0).SetBackend(phi::Backend::CPU);
kernel->InputAt(1).SetBackend(phi::Backend::CPU);
}

#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
Expand All @@ -130,7 +129,6 @@ PD_REGISTER_KERNEL(full_with_tensor_sr,
phi::dtype::complex<float>,
phi::dtype::complex<double>) {
kernel->InputAt(0).SetBackend(phi::Backend::CPU);
kernel->InputAt(1).SetBackend(phi::Backend::CPU);
}
#endif

Expand All @@ -147,6 +145,5 @@ PD_REGISTER_KERNEL(full_with_tensor_sr,
bool,
phi::dtype::float16) {
kernel->InputAt(0).SetBackend(phi::Backend::CPU);
kernel->InputAt(1).SetBackend(phi::Backend::CPU);
}
#endif
1 change: 0 additions & 1 deletion paddle/phi/kernels/xpu/full_kernel.cc
Original file line number Diff line number Diff line change
Expand Up @@ -171,5 +171,4 @@ PD_REGISTER_KERNEL(full_with_tensor,
bool,
phi::dtype::float16) {
kernel->InputAt(0).SetBackend(phi::Backend::CPU);
kernel->InputAt(1).SetBackend(phi::Backend::CPU);
}

0 comments on commit 4eb91a6

Please sign in to comment.