【PIR Dist Op Reg No.18】 reg lars_momentum

[enkilee]

PR types

Others

PR changes

Others

Description

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

[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]

/paddle/paddle/fluid/pybind/ops_api.cc:4721:12: error: ???eager_api_lars_momentum_??? was not declared in this scope

这里需要把lars_momentum_和lars_momentum加入到paddle/fluid/pir/dialect/op_generator/ops_api_gen.py中的NO_NEED_GEN_STATIC_ONLY_APIS列表中

[kangguangli]

可以这样修改试试，我本地可以通过编译及相关单测

[kangguangli]

@enkilee hello，之前的问题花了点时间，主要原因是执行器在构建Context时对optional类型变量的处理有一些问题，我在 #61314 进行了修复，该PR已经合入。不过还有一点修改需要处理下，你可以再修改提交下看看。