Fix O(#segments) per-comm memory probe that slows weight sync & training under expandable_segments - #2269

Fix O(#segments) per-comm memory probe that slows weight sync & training under expandable_segments#2269

yszhli wants to merge 1 commit into

THUDM:mainTHUDM/slime:mainfrom

yszhli:fix/per-comm-memory-probe-perfyszhli/slime:fix/per-comm-memory-probe-perfCopy head branch name to clipboard