add primitives: net.batch_all_to_all_v(), net.batch_all_gather_v() - Githubissues

microsoft / Tutel

Tutel MoE: An Optimized Mixture-of-Experts Implementation

MIT License

724 stars 93 forks source link

add primitives: net.batch_all_to_all_v(), net.batch_all_gather_v() #221

Closed ghostplant closed 10 months ago