-
Notifications
You must be signed in to change notification settings - Fork 5.7k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[AutoParallel] Add shard tensor and DistAttr api #55494
[AutoParallel] Add shard tensor and DistAttr api #55494
Conversation
你的PR提交成功,感谢你对开源项目的贡献! |
else: | ||
raise NotImplementedError( | ||
"The `paddle.distributed.shard_tensor` for static mode will be implemented later." | ||
) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
这个接口是不是只给动态图用就行了,动静会在这里统一吗
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
按线下讨论结论,咱们先按统一做
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM for docs,请提供相应的中文文档
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
感谢 |
* add shard tensor api * add DistAttr api * add unittest for coverage * fix process mesh sample code * fix checking error
* add shard tensor api * add DistAttr api * add unittest for coverage * fix process mesh sample code * fix checking error
* add shard tensor api * add DistAttr api * add unittest for coverage * fix process mesh sample code * fix checking error
PR types
New features
PR changes
APIs
Description
Pcard-73145
[AutoParallel] Add shard tensor and DistAttr api
添加基础api的初版实现,便于采用标准用法构建单元测试
TODO:相应api功能实现后续仍需要补全完善