Update distributed_device_mesh.rst

SwayamInSync · web-flow · commit eb3272c85924 · 2025-08-20T12:02:42.000+05:30
diff --git 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``.

Original file line number	Diff line number	Diff line change
`@@ -138,7 +138,7 @@ users would not need to manually create and manage shard group and replicate gro`
`138`	`138`	`# HSDP: MeshShape(2, 4)`
`139`	`139`	`mesh_2d = init_device_mesh("cuda", (2, 4), mesh_dim_names=("dp_replicate", "dp_shard"))`
`140`	`140`	`model = FSDP(`
`141`		`- ToyModel(), device_mesh=mesh_2d`
	`141`	`+ ToyModel(), mesh=mesh_2d`
`142`	`142`	`)`
`143`	`143`
`144`	`144`	Let's create a file named ``hsdp.py``.