DeepSpeed

5ca81678 - [zero] fix missed subclasses partitioning bug (#1135)

[zero] fix missed subclasses partitioning bug (#1135) * fix missed subclassed partitioning bug * fix on exit Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>