This repository has been archived by the owner on Sep 15, 2021. It is now read-only.
v0.3.0
v0.3.0
⚠ BREAKING CHANGE
BaguaBucketPy
andBaguaTensorPy
now require name.BaguaTensorPy
is created by passing pytorch tensor directly now.
Features
- Compatible mode for getting pytorch tensor info with Python interpreter 1534d23
- Support creating BaguaTensor by passing torch tensor directly (#19) 4306e94
- Support Reduction op selection (SUM, MAX, etc.) b1bf784
- Add
reduce
andallgather
python interface ff68a61 - Support
alltoall_v
(vector alltoall) (#14) a6fe110