Symbolic shape inference support for pd_op.split and builtin.split #62394

fty1777 · 2024-03-04T15:50:41Z

PR types

Others

PR changes

Others

Description

Add Infer Symbolic Shape for pd_op.split and builtin.split

paddle-bot · 2024-03-04T15:50:47Z

你的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.

CLAassistant · 2024-03-04T15:50:48Z

All committers have signed the CLA.

CLAassistant · 2024-03-04T15:50:49Z

Thank you for your submission! We really appreciate it. Like many open source projects, we ask that you sign our Contributor License Agreement before we can accept your contribution.
_{You have signed the CLA already but the status is still pending? Let us recheck it.}

paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/paddle_op_infer_sym.cc

paddle/fluid/pir/dialect/operator/ir/op_dialect.cc

zhangbopd

进一步完善单测并且贴一下结果吧

…pe/paddle_op_infer_sym.cc Co-authored-by: Bo Zhang <105368690+zhangbopd@users.noreply.github.com>

Co-authored-by: Bo Zhang <105368690+zhangbopd@users.noreply.github.com>

…or outputs

fty1777 · 2024-03-05T18:46:04Z

test/ir/pir/cinn/symbolic/test_op_infer_sym_shape.py

@@ -351,7 +351,7 @@ def test_eval_symbolic(self):
                np.testing.assert_equal(
                    sym_shape_str_list[j].find(self.expected[i][j]),
                    0,
-                    f'in case i,j = {i},{j}: output shape ({sym_shape_str_list[0]}) is not expected {(self.expected[i][j])}',
+                    f'in case i,j = {i},{j}: output shape ({sym_shape_str_list[j]}) is not expected {(self.expected[i][j])}',


在写unittest的时候发现一些failure log打印的东西不太对，修了一下。

paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/paddle_op_infer_sym.cc

…addlePaddle#62394) * WIP: builtin.split op infer sym shape * bug fix * Update paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/paddle_op_infer_sym.cc Co-authored-by: Bo Zhang <105368690+zhangbopd@users.noreply.github.com> * Update paddle/fluid/pir/dialect/operator/ir/op_dialect.cc Co-authored-by: Bo Zhang <105368690+zhangbopd@users.noreply.github.com> * Update paddle/fluid/pir/dialect/operator/ir/op_dialect.cc Co-authored-by: Bo Zhang <105368690+zhangbopd@users.noreply.github.com> * pd_op.split followed by builtin.split * pd_op.split infer sym shape bugfix and unittest; fix op infer sym error outputs * recover SplitWithNumOpInferSymbolicShape Unimplemented exception raising * code refinement * Rewrite PADDLE_ENFORCE * remove incorrect comments * Rewrite PADDLE_ENFORCE * Rewrite PADDLE_ENFORCE --------- Co-authored-by: Bo Zhang <105368690+zhangbopd@users.noreply.github.com>

luotao1 · 2024-03-13T13:04:41Z

hi, @fty1777

非常感谢你对飞桨的贡献，我们正在运营一个PFCC组织，会通过定期分享技术知识与发布开发者主导任务的形式持续为飞桨做贡献，详情可见 https://github.com/luotao1 主页说明。
如果你对PFCC有兴趣，请发送邮件至 ext_paddle_oss@baidu.com，我们会邀请你加入~

WIP: builtin.split op infer sym shape

0f4b509

paddle-bot bot added the contributor External developers label Mar 4, 2024

bug fix

c8ee33c