I'm working on MoE model simulation using astra-sim 2.0 and noticed that Chakra supports alltoall, but I can't find support for alltoallv (variable-length all-to-all).
Unlike regular alltoall with fixed send/recv sizes, alltoallv uses variable counts and offsets per rank, which is widely used in modern MoE models for dynamic token routing.
- ✅ Supported: alltoall
- ❌ Unclear: alltoallv
Could you clarify whether Chakra already supports alltoallv, or if it's planned for future support?
I'm working on MoE model simulation using astra-sim 2.0 and noticed that Chakra supports
alltoall, but I can't find support for alltoallv (variable-length all-to-all).Unlike regular alltoall with fixed send/recv sizes, alltoallv uses variable counts and offsets per rank, which is widely used in modern MoE models for dynamic token routing.
Could you clarify whether Chakra already supports alltoallv, or if it's planned for future support?