DeepSpeed

fix(zero3): use current_stream() instead of default_stream() for grad…

#7898

Merged