File tree Expand file tree Collapse file tree 3 files changed +3
-2
lines changed
iluvatar_gpu/kernels/cuda_kernels Expand file tree Collapse file tree 3 files changed +3
-2
lines changed 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/gru_kernel.cu " // NOLINT
16+ #include " paddle/phi/kernels/gpu/gru_kernel.h "
1717#include " paddle/phi/kernels/impl/gru_kernel_impl.h"
1818
1919PD_CUSTOM_KERNEL_REGISTER (
Original file line number Diff line number Diff 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
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/gru_kernel.cu " // NOLINT
16+ #include " paddle/phi/kernels/gpu/gru_kernel.h "
1717#include " paddle/phi/kernels/impl/gru_kernel_impl.h"
1818
1919PD_CUSTOM_KERNEL_REGISTER (
You can’t perform that action at this time.
0 commit comments