Skip to content

Commit ddd5c22

Browse files
algorithm1832YqGe585luotao1
authored
include .h file for gru_kernel (#2126)
Co-authored-by: Yuqiang Ge <143453447+YqGe585@users.noreply.github.com> Co-authored-by: Tao Luo <luotao02@baidu.com>
1 parent 4aa8bea commit ddd5c22

File tree

3 files changed

+3
-2
lines changed

3 files changed

+3
-2
lines changed

backends/iluvatar_gpu/kernels/cuda_kernels/gru_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/gru_kernel.cu" //NOLINT
16+
#include "paddle/phi/kernels/gpu/gru_kernel.h"
1717
#include "paddle/phi/kernels/impl/gru_kernel_impl.h"
1818

1919
PD_CUSTOM_KERNEL_REGISTER(

backends/metax_gpu/CMakeLists.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -199,6 +199,7 @@ file(
199199
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/gather_nd_kernel.cu
200200
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/gather_nd_grad_kernel.cu
201201
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/gaussian_kernel.cu
202+
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/gru_kernel.cu
202203
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/index_add_kernel.cu
203204
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/index_put_kernel.cu
204205
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/index_put_grad_kernel.cu

backends/metax_gpu/kernels/cuda_kernels/gru_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/gru_kernel.cu" //NOLINT
16+
#include "paddle/phi/kernels/gpu/gru_kernel.h"
1717
#include "paddle/phi/kernels/impl/gru_kernel_impl.h"
1818

1919
PD_CUSTOM_KERNEL_REGISTER(

0 commit comments

Comments
 (0)