Introduce to_gangs helper (#986)

facebookresearch · Jan 24, 2025 · cf5f7bb · cf5f7bb
1 parent d5f688a
commit cf5f7bb
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 7 deletions.
diff --git a/.github/workflows/release.yaml b/.github/workflows/release.yaml
@@ -69,16 +69,17 @@ jobs:
       release_type: ${{ inputs.release_type }}
       version_override: ${{ needs.process_version.outputs.version_override }}
 
-  build_doc:
-    name: Build documentation
-    needs: [process_version]
-    uses: ./.github/workflows/_build_doc.yaml
-    with:
-      version_override: ${{ needs.process_version.outputs.version_override }}
+#  build_doc:
+#    name: Build documentation
+#    needs: [process_version]
+#    uses: ./.github/workflows/_build_doc.yaml
+#    with:
+#      version_override: ${{ needs.process_version.outputs.version_override }}
 
   publish:
     name: Publish
-    needs: [build_wheels, build_doc]
+    needs: [build_wheels]
+#    needs: [build_wheels, build_doc]
     uses: ./.github/workflows/_publish.yaml
     with:
       release_type: ${{ inputs.release_type }}
diff --git a/src/fairseq2/gang.py b/src/fairseq2/gang.py
@@ -713,6 +713,12 @@ def fake_gangs(device: Device) -> Gangs:
     return Gangs(gang, gang, gang)
 
 
+def to_gangs(gang: Gang) -> Gangs:
+    fake_gang = FakeGang(device=gang.device)
+
+    return Gangs(gang, gang, fake_gang)
+
+
 def _setup_2D_mesh_gangs(
     root_gang: Gang,
     *,