[Fix][muon] fix assertion err when dim0 size < world_size#1611
Open
nil0x9 wants to merge 1 commit intoInternLM:mainfrom
Open
[Fix][muon] fix assertion err when dim0 size < world_size#1611nil0x9 wants to merge 1 commit intoInternLM:mainfrom
nil0x9 wants to merge 1 commit intoInternLM:mainfrom
