Skip to content

Commit 5f0c05a

Browse files
authored
Update distributed_device_mesh.rst
1 parent 619f184 commit 5f0c05a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

recipes_source/distributed_device_mesh.rst

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ Users can also easily manage the underlying process_groups/devices for multi-dim
3131
Why DeviceMesh is Useful
3232
------------------------
3333
DeviceMesh is useful when working with multi-dimensional parallelism (i.e. 3-D parallel) where parallelism composability is required. For example, when your parallelism solutions require both communication across hosts and within each host.
34-
The image above shows that we can create a 2D mesh that connects the devices within each host, and connects each device with its counterpart on the other hosts in a homogenous setup.
34+
The image above shows that we can create a 2D mesh that connects the devices within each host, and connects each device with its counterpart on the other hosts in a homogeneous setup.
3535

3636
Without DeviceMesh, users would need to manually set up NCCL communicators, cuda devices on each process before applying any parallelism, which could be quite complicated.
3737
The following code snippet illustrates a hybrid sharding 2-D Parallel pattern setup without :class:`DeviceMesh`.
@@ -150,7 +150,7 @@ Then, run the following `torch elastic/torchrun <https://pytorch.org/docs/stable
150150
151151
How to use DeviceMesh for your custom parallel solutions
152152
--------------------------------------------------------
153-
When working with large scale training, you might have more complex custom parallel training composition. For example, you may need to slice out submeshes for different parallelism solutions.
153+
When working with large scale training, you might have more complex custom parallel training composition. For example, you may need to slice out sub-meshes for different parallelism solutions.
154154
DeviceMesh allows users to slice child mesh from the parent mesh and re-use the NCCL communicators already created when the parent mesh is initialized.
155155

156156
.. code-block:: python

0 commit comments

Comments
 (0)