Fix a bug in DeepSpeedMLP (#4389)

Commit

1 year ago

Fix a bug in DeepSpeedMLP (#4389) Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com>