Skip to content

Commit 64d769a

Browse files
123wjrluotao1YqGe585
authored
【CUDA Kernel No.47-48】 算子Kernel修复 (#2178)
* fix: update include directive for deformable_conv_grad_kernel * fix: update include directive for deformable_conv_kernel_register * fix: update include directive for deformable_conv_grad_kernel * fix: add missing include for kernel registry in deformable_conv_grad_kernel_register * fix: add missing source files for deformable convolution kernels in CMakeLists --------- Co-authored-by: Tao Luo <[email protected]> Co-authored-by: Yuqiang Ge <[email protected]>
1 parent 982ed6f commit 64d769a

File tree

3 files changed

+5
-2
lines changed

3 files changed

+5
-2
lines changed

backends/metax_gpu/CMakeLists.txt

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -301,6 +301,8 @@ file(
301301
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/broadcast_kernel.cu
302302
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/decayed_adagrad_kernel.cu
303303
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/debug_tools_kernel.cu
304+
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/deformable_conv_kernel.cu
305+
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/deformable_conv_grad_kernel.cu
304306
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/cumprod_grad_kernel.cu
305307
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/cumprod_kernel.cu
306308
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/crop_grad_kernel.cu

backends/metax_gpu/kernels/cuda_kernels/deformable_conv_grad_kernel_register.cu

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,8 @@
1212
// See the License for the specific language governing permissions and
1313
// limitations under the License.
1414

15-
#include "paddle/phi/kernels/gpu/deformable_conv_grad_kernel.cu" // NOLINT
15+
#include "paddle/phi/core/kernel_registry.h"
16+
#include "paddle/phi/kernels/deformable_conv_grad_kernel.h"
1617

1718
PD_CUSTOM_KERNEL_REGISTER(deformable_conv_grad,
1819
metax_gpu,

backends/metax_gpu/kernels/cuda_kernels/deformable_conv_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/deformable_conv_kernel.cu" // NOLINT
16+
#include "paddle/phi/kernels/deformable_conv_kernel.h"
1717

1818
PD_CUSTOM_KERNEL_REGISTER(deformable_conv,
1919
metax_gpu,

0 commit comments

Comments
 (0)