flashinfer.comm.MoeAlltoAll.checkpoint_restore¶
- MoeAlltoAll.checkpoint_restore(comm_backend: CommBackend) None¶
Remap MNNVL handles after restore; repeated calls are no-ops.
- Parameters:
comm_backend (CommBackend) – Communication backend used to recreate and exchange MNNVL memory handles. It must have the same rank and world size as the original allocation.