[Typing][C-12] Add type annotations for python/paddle/distributed/communication/all_gather.py

[megemini]

PR Category

User Experience

PR Types

Improvements

Description

类型标注：

• python/paddle/distributed/communication/all_gather.py

Related links

• [Type Hints] 为公开 API 标注类型提示信息 #65008

@SigureMo @megemini

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

[megemini]

补充：

是否可以在 _typing 中引入 _libpaddle.pyi 文件？

标注 distributed 的时候，发现有很多是直接使用 c++ 绑定到 paddle.base.libpaddle 中的类 ～ 之前也遇到过 ～

是否可以引入此文件，将需要用到的 paddle.base.libpaddle 放到 _libpaddle.pyi 中，并作为 Protocol？

[megemini]

Update 20240716

paddle.base.libpaddle 与 paddle.base.core 貌似是等价的 ～

尝试去掉 _typing._libpaddle ，在 paddle.base.core.pyi 中增加 ～

[megemini]

在考虑，要不要搞个 core.prototype.pyi ...

p.s. 有个问题是，libpaddle 里面的东西是跟随编译环境生成的，比如 DistFleetWrapper ，需要分布式编译选项才有，因此，如果只是根据 libpaddle 内容生成 stub，东西可能不全 ... ...

[SigureMo]

在考虑，要不要搞个 core.prototype.pyi ...

可以呀

p.s. 有个问题是，libpaddle 里面的东西是跟随编译环境生成的，比如 DistFleetWrapper ，需要分布式编译选项才有，因此，如果只是根据 libpaddle 内容生成 stub，东西可能不全 ... ...

如果用户在用开分布式选项的包，那么就一定会有这个 stub，如果用的是没有编分布式的包，那么运行时也不会访问，没有也没什么关系

如果是 CI 检查的话，可以考虑整个 Any 啥的 Fallback 机制吧，感觉也没啥更好的办法

[SigureMo]

LGTMeow