Function _sizes_all_same

deepspeed/runtime/zero/mics_utils.py:197–203

all groups have same length

(groups)

def _sizes_all_same(groups):
    """all groups have same length"""
    all_same = True
    for g in groups:
        if len(g) != len(groups[0]):
            return False
    return all_same