[QUESTION] `save_checkpoint` with `expert_tensor_parallel_size`
Open
community-request
module: moe
question
- Dominant language
- Python
- Stars
- 17.9k
- Forks
- 4.5k
- Avg merge
- 4d 6h
- Merged PRs (30d)
- 271
Description
**Your question**
How to save a sharded checkpoint when using MoE parallel folding -- specifically, when `ETP` != `TP`? `save_checkpoint` only supports `PP`, `TP`, and `EP` (as in `expert_model_parallel` not `expert_tensor_parallel`)?
Contributor guide
Assessment
This issue has not been assessed yet.