【PIR Dist Op Reg No.2】 reg distributed_push_sparse

[enkilee]

PR types

Others

PR changes

Others

Description

• 【PIR】PIR下的分布式算子注册 #60436
  No.2 distributed_push_sparse

[paddle-bot]

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

[xingmingyyj]

建议补充一下单测
现在op注册时的output名称是output

  output : Tensor[](output){ids.size()}

但是op_compat里面的配置是out，

- op : distributed_push_sparse
  inputs :
    {ids : Ids, shows : Shows, clicks: Clicks}
  outputs :
    out : Outputs

可以设置一下单测，确认一下能不能正确翻译

[zyfncg]

这个output不需要设置dim信息吗？

[kangguangli]

这个output不需要设置dim信息吗？

原本的InferShape里并没有相关代码，这类Op一般是通过pass插入，不是正常组网得到的，我觉得还是不加为好。