Skip to content

Commit 0c0c07a

Browse files
authored
【CUDA Kernel No.100】roi_align算子Kernel修复 (#2005)
1 parent 2f8f6e3 commit 0c0c07a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

backends/iluvatar_gpu/kernels/cuda_kernels/roi_align_kernel_register.cc

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/roi_align_kernel.cu" // NOLINT
16+
#include "paddle/phi/kernels/roi_align_kernel.h"
1717

1818
PD_CUSTOM_KERNEL_REGISTER(
1919
roi_align, iluvatar_gpu, ALL_LAYOUT, phi::RoiAlignKernel, float) {}

0 commit comments

Comments
 (0)