DeepSpeed
Stage_1_and_2.py: fix assert for reduce_scatter configurations combinations
#4964
Merged
Merged