-
Notifications
You must be signed in to change notification settings - Fork 5.6k
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
Completes basic dtypes for all_reduce api in eager mode #45440
Conversation
你的PR提交成功,感谢你对开源项目的贡献! |
✅ This PR's description meets the template requirements! |
ad823f8
to
ce7f856
Compare
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.
To add support of bool
type, should we modify c++ backend?
python/paddle/fluid/tests/unittests/test_collective_allreduce_api.py
Outdated
Show resolved
Hide resolved
Maybe we have supported |
0760773
to
2603cd2
Compare
2603cd2
to
9612f78
Compare
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
PR types
New features
PR changes
OPs
Describe
This pr completes the basic function of communication framework, support various data types.
all_reduce
in eager mode.通信框架功能进一步补全,通信操作支持传输丰富的数据类型。
all_reduce
操作,支持动态图场景下使用NCCL、GLOO后端进行int8、uint8、bool类型的数据传输。all_reduce
的测试。