DeepSpeed

fix: Fix undefined variable in _create_expert_data_and_model_parallel and make it easier to understand

#1826

Merged