refactor: 优化参数传递，清理导入样式

2026-04-03 22:06:32 +08:00
parent 3a7d98a950
commit 0852b852f8
51 changed files with 299 additions and 434 deletions
@@ -1,14 +1,13 @@
+from astrai.parallel.module import ColumnParallelLinear, RowParallelLinear
 from astrai.parallel.setup import (
-    get_world_size,
-    get_rank,
    get_current_device,
+    get_rank,
+    get_world_size,
    only_on_rank,
    setup_parallel,
    spawn_parallel_fn,
 )

-from astrai.parallel.module import RowParallelLinear, ColumnParallelLinear
-
 __all__ = [
    "get_world_size",
    "get_rank",