Update sharded_moe.py to support top2 gate with Tutel#6948

xenshinu wants to merge 4 commits intodeepspeedai:masterdeepspeedai/DeepSpeed:masterfrom

xenshinu:patch-1xenshinu/DeepSpeed:patch-1Copy head branch name to clipboard