DeepSpeed

aed599b4 - Fix assertion to run pipeline engine with a compiled module (#5197)

Fix assertion to run pipeline engine with a compiled module (#5197) This PR fixes assertion in the pipeline engine to compute a compile module with pipeline parallelism.