Skip to content
Open
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@
// limitations under the License.

#include "paddle/phi/core/kernel_registry.h"
#include "paddle/phi/kernels/gpu/repeat_interleave_grad_kernel.cu" // NOLINT
#include "paddle/phi/kernels/repeat_interleave_grad_kernel.h"

PD_CUSTOM_KERNEL_REGISTER(repeat_interleave_with_tensor_index_grad,
iluvatar_gpu,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@
// limitations under the License.

#include "paddle/phi/core/kernel_registry.h"
#include "paddle/phi/kernels/gpu/repeat_interleave_kernel.cu" // NOLINT
#include "paddle/phi/kernels/repeat_interleave_kernel.h"

PD_CUSTOM_KERNEL_REGISTER(repeat_interleave,
iluvatar_gpu,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@
// limitations under the License.

#include "paddle/phi/core/kernel_registry.h"
#include "paddle/phi/kernels/gpu/rmsprop_kernel.cu" // NOLINT
#include "paddle/phi/kernels/rmsprop_kernel.h"

PD_CUSTOM_KERNEL_REGISTER(rmsprop,
iluvatar_gpu,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@
// limitations under the License.

#include "paddle/phi/core/kernel_registry.h"
#include "paddle/phi/kernels/gpu/roi_align_grad_kernel.cu" // NOLINT
#include "paddle/phi/kernels/roi_align_grad_kernel.h"

PD_CUSTOM_KERNEL_REGISTER(
roi_align_grad, iluvatar_gpu, ALL_LAYOUT, phi::RoiAlignGradKernel, float) {}
9 changes: 4 additions & 5 deletions backends/metax_gpu/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -124,11 +124,10 @@ file(
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/funcs/math/*.cu
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/funcs/eigen/*.cu
# kernels/gpu
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/partial_send_kernel.cu
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

不要删除这些文件。否则会导致单测报错

${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/partial_recv_kernel.cu
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/partial_concat_grad_kernel.cu
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/partial_allgather_kernel.cu
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/ctc_align_kernel.cu
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/rmsprop_kernel.cu
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/repeat_interleave_kernel.cu
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/repeat_interleave_grad_kernel.cu
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/quantize_linear_kernel.cu
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/yolo_box_head_kernel.cu
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/stft_grad_kernel.cu
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/soft_relu_grad_kernel.cu
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,6 @@

#include "paddle/phi/core/kernel_registry.h"
#include "paddle/phi/kernels/gpu/reduce_kernel.cu" // NOLINT
#include "paddle/phi/kernels/reduce_kernel.h"

PD_CUSTOM_KERNEL_REGISTER(reduce,
metax_gpu,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@
// limitations under the License.

#include "paddle/phi/core/kernel_registry.h"
#include "paddle/phi/kernels/gpu/repeat_interleave_grad_kernel.cu" // NOLINT
#include "paddle/phi/kernels/repeat_interleave_grad_kernel.h"

PD_CUSTOM_KERNEL_REGISTER(repeat_interleave_with_tensor_index_grad,
metax_gpu,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@
// limitations under the License.

#include "paddle/phi/core/kernel_registry.h"
#include "paddle/phi/kernels/gpu/repeat_interleave_kernel.cu" // NOLINT
#include "paddle/phi/kernels/repeat_interleave_kernel.h"

PD_CUSTOM_KERNEL_REGISTER(repeat_interleave,
metax_gpu,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@
// limitations under the License.

#include "paddle/phi/core/kernel_registry.h"
#include "paddle/phi/kernels/gpu/rmsprop_kernel.cu" // NOLINT
#include "paddle/phi/kernels/rmsprop_kernel.h"

PD_CUSTOM_KERNEL_REGISTER(rmsprop,
metax_gpu,
Expand Down
Loading