transformers
f73a4db3
- Inclusion of process_group in the gather_full_tensor function in tensor_parallel.py (#43932)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
145 days ago
Inclusion of process_group in the gather_full_tensor function in tensor_parallel.py (#43932) * Bug fix on gather_full_tensor for upstreaming. * Made correct usage of process_group in all_gather now.
References
#43932 - Inclusion of process_group in the gather_full_tensor function in tensor_parallel.py
Author
quic-meetkuma
Parents
3e4b8d63
Loading