transformers
24ab0dcc
- [distributed] Wire resolve_parallel_plan into apply_tensor_parallel
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
106 days ago
[distributed] Wire resolve_parallel_plan into apply_tensor_parallel Replace exclusive SP|EP|TP plan selection with merged plans when tp_plan is unset. Add distributed test for TP+EP merged expert sharding.
References
#46292 - sp + ep training / tp + ep inference
Author
3outeille
Committer
3outeille
Parents
5f4729be
Loading