search

PaddlePaddle / Paddle

PArallel Distributed Deep LEarning: Machine Learning Framework from Industrial Practice （『飞桨』核心框架，深度学习&机器学习高性能单机、分布式训练和跨平台部署）
http://www.paddlepaddle.org/
Apache License 2.0
21.66k stars 5.44k forks source link

[CINN]Add InferSymbolicShape of fpn ops #63947

Closed yulangz closed 1 week ago

yulangz commented 3 weeks ago

PR Category

CINN

PR Types

Bug fixes

Description

Add DistributeFpnProposalsOpInferSymbolicShape; Add RoiAlignOpInferSymbolicShape; fix GatherOpInferSymbolicShape when axis input is tensor;

paddle-bot[bot] commented 3 weeks ago

你的PR提交成功，感谢你对开源项目的贡献! 请关注后续CI自动化测试结果，详情请参考Paddle-CI手册。 Your PR has been submitted. Thanks for your contribution! Please wait for the result of CI firstly. See Paddle CI Manual for details.

yulangz commented 3 weeks ago

对应单测： layercase/sublayer1000/Det_cases/cascade_rcnn_cascade_mask_rcnn_r50_vd_fpn_ssld_1x_coco/SIR_145.py layercase/sublayer1000/Det_cases/mask_rcnn_mask_rcnn_r101_fpn_1x_coco/SIR_106.py layercase/sublayer1000/Det_cases/mask_rcnn_mask_rcnn_r50_fpn_1x_coco/SIR_106.py layercase/sublayer1000/Det_cases/mask_rcnn_mask_rcnn_r50_fpn_2x_coco/SIR_109.py layercase/sublayer1000/Det_cases/mask_rcnn_mask_rcnn_x101_vd_64x4d_fpn_1x_coco/SIR_108.py layercase/sublayer1000/Det_cases/mask_rcnn_mask_rcnn_x101_vd_64x4d_fpn_2x_coco/SIR_108.py layercase/sublayer1000/Det_cases/res2net_mask_rcnn_res2net50_vd_26w_4s_fpn_2x_coco/SIR_132.py

目前还存在与 Gather 算子实现相关的问题，报错如下： image

由于 Gather 是子图的最后一个步骤，如果屏蔽掉 Gather，所有的输出都是正确的，如下所示： image