The paper reports that the modified Mamba-2 block is tensor-parallel friendly and reduces synchronization points per block by half. It also describes sequence/context parallelism in which devices process contiguous sequence chunks and pass recurrent states between workers, with communication linear in the number of workers. · CiteArk
The paper reports that the modified Mamba-2 block is tensor-parallel friendly and reduces synchronization points per block by half. It also describes sequence/context parallelism in which devices process contiguous sequence chunks and pass recurrent states between workers, with communication linear in the number of workers.
来源:paper_markdown:Introduction and Sections 8.1–8.2, PDF pages 3 and 27–28