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/c_scatter_kernel.cu " // NOLINT
16+ #include " paddle/phi/kernels/gpu/c_scatter_kernel.h "
1717
1818PD_CUSTOM_KERNEL_REGISTER (c_scatter,
1919 iluvatar_gpu,
Original file line number Diff line number Diff line change @@ -158,6 +158,7 @@ file(
158158 ${PADDLE_SOURCE_DIR} /paddle/phi/kernels/gpu/c_concat_kernel.cu
159159 ${PADDLE_SOURCE_DIR} /paddle/phi/kernels/gpu/c_embedding_kernel.cu
160160 ${PADDLE_SOURCE_DIR} /paddle/phi/kernels/gpu/c_embedding_grad_kernel.cu
161+ ${PADDLE_SOURCE_DIR} /paddle/phi/kernels/gpu/c_scatter_kernel.cu
161162 ${PADDLE_SOURCE_DIR} /paddle/phi/kernels/gpu/cast_kernel.cu
162163 ${PADDLE_SOURCE_DIR} /paddle/phi/kernels/gpu/clip_kernel.cu
163164 ${PADDLE_SOURCE_DIR} /paddle/phi/kernels/gpu/clip_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/c_scatter_kernel.cu " // NOLINT
16+ #include " paddle/phi/kernels/gpu/c_scatter_kernel.h "
1717
1818PD_CUSTOM_KERNEL_REGISTER (c_scatter,
1919 metax_gpu,
You can’t perform that action at this time.
0 commit comments