Skip to content

Commit b39fff0

Browse files
authored
【CUDA Kernel No.43】correlation算子Kernel修复 (#2098)
1 parent 08c3a22 commit b39fff0

File tree

3 files changed

+3
-2
lines changed

3 files changed

+3
-2
lines changed

backends/iluvatar_gpu/kernels/cuda_kernels/correlation_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

1515
#include "paddle/phi/core/kernel_registry.h"
16-
#include "paddle/phi/kernels/gpu/correlation_kernel.cu" //NOLINT
16+
#include "paddle/phi/kernels/gpu/correlation_kernel.h"
1717

1818
PD_CUSTOM_KERNEL_REGISTER(correlation,
1919
iluvatar_gpu,

backends/metax_gpu/CMakeLists.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -284,6 +284,7 @@ file(
284284
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/reduce_kernel.cu
285285
${PADDLE_SOURCE_DIR}/paddle/phi/backends/dynload/cusolver.cc
286286
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/collect_fpn_proposals_kernel.cu
287+
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/correlation_kernel.cu
287288
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/clip_by_norm_kernel.cu
288289
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/check_numerics_kernel.cu
289290
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/comm_init_all_kernel.cu

backends/metax_gpu/kernels/cuda_kernels/correlation_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

1515
#include "paddle/phi/core/kernel_registry.h"
16-
#include "paddle/phi/kernels/gpu/correlation_kernel.cu" //NOLINT
16+
#include "paddle/phi/kernels/gpu/correlation_kernel.h"
1717

1818
PD_CUSTOM_KERNEL_REGISTER(correlation,
1919
metax_gpu,

0 commit comments

Comments
 (0)