From eb3272c859249687bc29e64cecf9c9a4584fff7e Mon Sep 17 00:00:00 2001 From: Swayam Date: Wed, 20 Aug 2025 12:02:42 +0530 Subject: [PATCH] Update distributed_device_mesh.rst --- recipes_source/distributed_device_mesh.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/recipes_source/distributed_device_mesh.rst b/recipes_source/distributed_device_mesh.rst index 3a04b8de4bf..cc982c11790 100644 --- a/recipes_source/distributed_device_mesh.rst +++ b/recipes_source/distributed_device_mesh.rst @@ -138,7 +138,7 @@ users would not need to manually create and manage shard group and replicate gro # HSDP: MeshShape(2, 4) mesh_2d = init_device_mesh("cuda", (2, 4), mesh_dim_names=("dp_replicate", "dp_shard")) model = FSDP( - ToyModel(), device_mesh=mesh_2d + ToyModel(), mesh=mesh_2d ) Let's create a file named ``hsdp.py``.