DeepSpeed

5fb4256a - fix: fix undefined variable in MoE top2gating (#1827)

fix: fix undefined variable in MoE top2gating (#1827) Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com> Co-authored-by: Jeff Rasley <jerasley@microsoft.com>