File tree Expand file tree Collapse file tree 4 files changed +2
-2
lines changed
Expand file tree Collapse file tree 4 files changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -357,6 +357,7 @@ file(
357357 ${PADDLE_SOURCE_DIR} /paddle/phi/kernels/gpu/moe_unpermute_kernel.cu
358358 ${PADDLE_SOURCE_DIR} /paddle/phi/kernels/gpu/momentum_kernel.cu
359359 ${PADDLE_SOURCE_DIR} /paddle/phi/kernels/gpu/mp_allreduce_sum_kernel.cu
360+ ${PADDLE_SOURCE_DIR} /paddle/phi/kernels/gpu/multiclass_nms3_kernel.cu
360361 ${PADDLE_SOURCE_DIR} /paddle/phi/kernels/gpu/multiplex_grad_kernel.cu
361362 ${PADDLE_SOURCE_DIR} /paddle/phi/kernels/gpu/nonzero_kernel.cu
362363 ${PADDLE_SOURCE_DIR} /paddle/phi/kernels/gpu/pad3d_kernel.cu
Original file line number Diff line number Diff line change 1313// limitations under the License.
1414
1515#include " paddle/phi/core/kernel_registry.h"
16- #include " paddle/phi/kernels/gpu/multiclass_nms3_kernel.cu" // NOLINT
1716#include " paddle/phi/kernels/multiclass_nms3_kernel.h"
1817
1918PD_CUSTOM_KERNEL_REGISTER (multiclass_nms3,
Original file line number Diff line number Diff line change @@ -252,6 +252,7 @@ file(
252252 ${PADDLE_SOURCE_DIR} /paddle/phi/kernels/gpu/mean_all_grad_kernel.cu
253253 ${PADDLE_SOURCE_DIR} /paddle/phi/kernels/gpu/multiplex_kernel.cu
254254 ${PADDLE_SOURCE_DIR} /paddle/phi/kernels/gpu/multiplex_grad_kernel.cu
255+ ${PADDLE_SOURCE_DIR} /paddle/phi/kernels/gpu/multiclass_nms3_kernel.cu
255256 ${PADDLE_SOURCE_DIR} /paddle/phi/kernels/gpu/mp_allreduce_sum_kernel.cu
256257 ${PADDLE_SOURCE_DIR} /paddle/phi/kernels/gpu/pow2_decay_with_linear_warmup_kernel.cu
257258 ${PADDLE_SOURCE_DIR} /paddle/phi/kernels/gpu/sigmoid_cross_entropy_with_logits_grad_kernel.cu
Original file line number Diff line number Diff line change 1313// limitations under the License.
1414
1515#include " paddle/phi/core/kernel_registry.h"
16- #include " paddle/phi/kernels/gpu/multiclass_nms3_kernel.cu" // NOLINT
1716#include " paddle/phi/kernels/multiclass_nms3_kernel.h"
1817
1918PD_CUSTOM_KERNEL_REGISTER (multiclass_nms3,
You can’t perform that action at this time.
0 commit comments