chore: 更新项目名称

2026-03-31 09:34:11 +08:00
parent 780b9e1855
commit 2e009cf59a
51 changed files with 118 additions and 118 deletions
@@ -0,0 +1,21 @@
+from astrai.parallel.setup import (
+    get_world_size,
+    get_rank,
+    get_current_device,
+    only_on_rank,
+    setup_parallel,
+    spawn_parallel_fn,
+)
+
+from astrai.parallel.module import RowParallelLinear, ColumnParallelLinear
+
+__all__ = [
+    "get_world_size",
+    "get_rank",
+    "get_current_device",
+    "only_on_rank",
+    "setup_parallel",
+    "spawn_parallel_fn",
+    "RowParallelLinear",
+    "ColumnParallelLinear",
+]