Skip to content

Commit 3d3a0ce

Browse files
authored
fix include for multiplex_grad_kernel (PaddlePaddle#2204)
1 parent cd2e016 commit 3d3a0ce

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

backends/iluvatar_gpu/kernels/cuda_kernels/multiplex_grad_kernel_register.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
// limitations under the License.
1414
// clang-format off
1515
#include "paddle/phi/core/kernel_registry.h"
16-
#include "paddle/phi/kernels/gpu/multiplex_grad_kernel.cu" // NOLINT
16+
#include "paddle/phi/kernels/multiplex_grad_kernel.h" // NOLINT
1717
#include "paddle/phi/kernels/multinomial_kernel.h"
1818
// clang-format on
1919
PD_CUSTOM_KERNEL_REGISTER(multiplex_grad,

backends/metax_gpu/kernels/cuda_kernels/multiplex_grad_kernel_register.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414
// limitations under the License.
1515
// clang-format off
1616
#include "paddle/phi/core/kernel_registry.h"
17-
#include "paddle/phi/kernels/gpu/multiplex_grad_kernel.cu" // NOLINT
17+
#include "paddle/phi/kernels/multiplex_grad_kernel.h" // NOLINT
1818
#include "paddle/phi/kernels/multinomial_kernel.h"
1919
// clang-format on
2020
PD_CUSTOM_KERNEL_REGISTER(multiplex_grad,

0 commit comments

Comments
 (0)