Skip to content

Commit e9431ab

Browse files
authored
【CUDA Kernel No.38】add class_center_sample_kernel.h -part (PaddlePaddle#76374)
* add class_center_sample_kernel.h * Update class_center_sample_kernel.cu * Update class_center_sample_kernel.cu * Update class_center_sample_kernel.cu * fix codestyle
1 parent 3edeb00 commit e9431ab

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

paddle/phi/kernels/cpu/class_center_sample_kernel.cc

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
#include "paddle/phi/backends/cpu/cpu_context.h"
2020
#include "paddle/phi/core/generator.h"
2121
#include "paddle/phi/core/kernel_registry.h"
22+
#include "paddle/phi/kernels/class_center_sample_kernel.h"
2223

2324
namespace phi {
2425

paddle/phi/kernels/gpu/class_center_sample_kernel.cu

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@ namespace cub = hipcub;
3232
#include "paddle/phi/common/memory_utils.h"
3333
#include "paddle/phi/core/enforce.h"
3434
#include "paddle/phi/core/tensor_utils.h"
35+
#include "paddle/phi/kernels/class_center_sample_kernel.h"
3536

3637
#if defined(PADDLE_WITH_NCCL) || defined(PADDLE_WITH_RCCL)
3738
#include "paddle/common/flags.h"

0 commit comments

Comments
 (0)