[pipeline] add stage manager (hpcaitech#4093)

* [pipeline] add stage manager * [test] add pipeline stage manager test * [pipeline] add docstring for stage manager
ver217 · Aug 15, 2023 · b902be2 · b902be2
1 parent 3c2fbe2
commit b902be2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_pipeline/test_stage_manager.py b/tests/test_pipeline/test_stage_manager.py
@@ -21,7 +21,7 @@ def check_stage_manager():
         1: [0, 1],
         2: [2, 3],
         3: [2, 3],
-    }   
+    }
     pg_mesh = ProcessGroupMesh(DP_SIZE, PP_SIZE)
     stage_manager = PipelineStageManager(pg_mesh, PP_DIM)
     rank = dist.get_rank()