Skip to content

Commit 7e020a0

Browse files
authored
fix (#2207)
1 parent a71a4bc commit 7e020a0

File tree

9 files changed

+12
-7
lines changed

9 files changed

+12
-7
lines changed

backends/iluvatar_gpu/CMakeLists.txt

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -125,6 +125,10 @@ file(
125125
${PADDLE_SOURCE_DIR}/paddle/phi/backends/dynload/cublasLt.cc
126126
${PADDLE_SOURCE_DIR}/paddle/phi/backends/dynload/cufft.cc
127127
# kernels/gpu
128+
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/legacy/gpu/expand_modality_expert_id_kernel.cu
129+
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/legacy/gpu/cal_aux_loss_kernel.cu
130+
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/legacy/gpu/cal_aux_loss_grad_kernel.cu
131+
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/yolo_box_post_kernel.cu
128132
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/spectral_norm_grad_kernel.cu
129133
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/spectral_norm_kernel.cu
130134
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/abs_grad_kernel.cu

backends/iluvatar_gpu/kernels/cuda_kernels/yolo_box_post_kernel_register.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
#include "paddle/phi/core/kernel_registry.h"
1616
#include "paddle/phi/kernels/funcs/math_function.h"
1717
#include "paddle/phi/kernels/funcs/yolo_box_util.h"
18-
#include "paddle/phi/kernels/gpu/yolo_box_post_kernel.cu" //NOLINT
18+
#include "paddle/phi/kernels/gpu/yolo_box_post_kernel.h"
1919
PD_CUSTOM_KERNEL_REGISTER(
2020
yolo_box_post, iluvatar_gpu, ALL_LAYOUT, phi::YoloBoxPostKernel, float) {
2121
kernel->OutputAt(0).SetDataType(phi::DataType::FLOAT32);

backends/iluvatar_gpu/kernels/ernie_core/expand_modality_expert_id_kernel_register.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
// See the License for the specific language governing permissions and
1313
// limitations under the License.
1414
#include "paddle/phi/core/kernel_registry.h"
15-
#include "paddle/phi/kernels/legacy/gpu/expand_modality_expert_id_kernel.cu" // NOLINT
15+
#include "paddle/phi/kernels/legacy/gpu/expand_modality_expert_id_kernel.h"
1616

1717
PD_CUSTOM_KERNEL_REGISTER(expand_modality_expert_id,
1818
iluvatar_gpu,

backends/metax_gpu/CMakeLists.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -124,6 +124,7 @@ file(
124124
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/funcs/math/*.cu
125125
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/funcs/eigen/*.cu
126126
# kernels/gpu
127+
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/yolo_box_post_kernel.cu
127128
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/spectral_norm_grad_kernel.cu
128129
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/spectral_norm_kernel.cu
129130
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/activation_kernel.cu

backends/metax_gpu/kernels/cuda_kernels/yolo_box_post_kernel_register.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
#include "paddle/phi/core/kernel_registry.h"
1616
#include "paddle/phi/kernels/funcs/math_function.h"
1717
#include "paddle/phi/kernels/funcs/yolo_box_util.h"
18-
#include "paddle/phi/kernels/gpu/yolo_box_post_kernel.cu" //NOLINT
18+
#include "paddle/phi/kernels/gpu/yolo_box_post_kernel.h"
1919
PD_CUSTOM_KERNEL_REGISTER(
2020
yolo_box_post, metax_gpu, ALL_LAYOUT, phi::YoloBoxPostKernel, float) {
2121
kernel->OutputAt(0).SetDataType(phi::DataType::FLOAT32);

backends/metax_gpu/kernels/ernie_core/cal_aux_loss_grad_kernel_register.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
// See the License for the specific language governing permissions and
1313
// limitations under the License.
1414
#include "paddle/phi/core/kernel_registry.h"
15-
#include "paddle/phi/kernels/legacy/gpu/cal_aux_loss_grad_kernel.cu" //NOLINT
15+
#include "paddle/phi/kernels/legacy/gpu/cal_aux_loss_grad_kernel.h"
1616

1717
PD_CUSTOM_KERNEL_REGISTER(cal_aux_loss_grad,
1818
metax_gpu,

backends/metax_gpu/kernels/ernie_core/cal_aux_loss_kernel_register.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
// See the License for the specific language governing permissions and
1313
// limitations under the License.
1414
#include "paddle/phi/core/kernel_registry.h"
15-
#include "paddle/phi/kernels/legacy/gpu/cal_aux_loss_kernel.cu" //NOLINT
15+
#include "paddle/phi/kernels/legacy/gpu/cal_aux_loss_kernel.h"
1616

1717
PD_CUSTOM_KERNEL_REGISTER(cal_aux_loss,
1818
metax_gpu,

backends/metax_gpu/kernels/ernie_core/expand_modality_expert_id_kernel_register.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
// See the License for the specific language governing permissions and
1313
// limitations under the License.
1414
#include "paddle/phi/core/kernel_registry.h"
15-
#include "paddle/phi/kernels/legacy/gpu/expand_modality_expert_id_kernel.cu" //NOLINT
15+
#include "paddle/phi/kernels/legacy/gpu/expand_modality_expert_id_kernel.h"
1616

1717
PD_CUSTOM_KERNEL_REGISTER(expand_modality_expert_id,
1818
metax_gpu,

backends/metax_gpu/kernels/ernie_core/register_build_src_rank_and_local_expert_id_kernel.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
// See the License for the specific language governing permissions and
1313
// limitations under the License.
1414
#include "paddle/phi/core/kernel_registry.h"
15-
#include "paddle/phi/kernels/legacy/gpu/ext_build_src_rank_and_local_expert_id_kernel.cu" //NOLINT
15+
#include "paddle/phi/kernels/legacy/gpu/ext_build_src_rank_and_local_expert_id_kernel.h"
1616
PD_CUSTOM_KERNEL_REGISTER(build_src_rank_and_local_expert_id,
1717
metax_gpu,
1818
ALL_LAYOUT,

0 commit comments

Comments
 (0)