combine **inplace API
todo opened this issue · comments
bagua/bagua/torch_api/communication.py
Lines 1062 to 1067 in cd499b8
Bagua Speeds up PyTorch
todo opened this issue · comments
bagua/bagua/torch_api/communication.py
Lines 1062 to 1067 in cd499b8
TODO
comment in cd499b8. It's been assigned to @wangraying because they committed the code.