add shared expert cases

danielvegamyhre · danielvegamyhre · commit 7eb6bfd9d5f7 · 2025-08-29T13:03:39.000-07:00
diff --git a/test/prototype/moe_training/test_fsdp.py b/test/prototype/moe_training/test_fsdp.py
@@ -78,7 +78,7 @@ def device_mesh_1d() -> DeviceMesh:
     "target_fqns",
     [
         ["experts"],
-        ["does.not.exist"],
+        ["experts,shared_experts"],
     ],
 )
 @pytest.mark.parametrize("compile", [False, True])
diff --git a/test/prototype/moe_training/test_fsdp_tp.py b/test/prototype/moe_training/test_fsdp_tp.py
@@ -103,7 +103,7 @@ def device_mesh_2d() -> DeviceMesh:
     "target_fqns",
     [
         ["experts"],
-        ["does.not.exist"],
+        ["experts,shared_experts"],
     ],
 )
 @pytest.mark.parametrize("compile", [False, True])
diff --git a/test/prototype/moe_training/test_tp.py b/test/prototype/moe_training/test_tp.py
@@ -96,7 +96,7 @@ def device_mesh_1d() -> DeviceMesh:
     "target_fqns",
     [
         ["experts"],
-        ["does.not.exist"],
+        ["experts,shared_experts"],
     ],
 )
 @pytest.mark.parametrize("compile", [False, True])

Original file line number	Diff line number	Diff line change
`@@ -78,7 +78,7 @@ def device_mesh_1d() -> DeviceMesh:`
`78`	`78`	`"target_fqns",`
`79`	`79`	`[`
`80`	`80`	`["experts"],`
`81`		`- ["does.not.exist"],`
	`81`	`+ ["experts,shared_experts"],`
`82`	`82`	`],`
`83`	`83`	`)`
`84`	`84`	`@pytest.mark.parametrize("compile", [False, True])`
Original file line number	Diff line number	Diff line change
`@@ -103,7 +103,7 @@ def device_mesh_2d() -> DeviceMesh:`
`103`	`103`	`"target_fqns",`
`104`	`104`	`[`
`105`	`105`	`["experts"],`
`106`		`- ["does.not.exist"],`
	`106`	`+ ["experts,shared_experts"],`
`107`	`107`	`],`
`108`	`108`	`)`
`109`	`109`	`@pytest.mark.parametrize("compile", [False, True])`
Original file line number	Diff line number	Diff line change
`@@ -96,7 +96,7 @@ def device_mesh_1d() -> DeviceMesh:`
`96`	`96`	`"target_fqns",`
`97`	`97`	`[`
`98`	`98`	`["experts"],`
`99`		`- ["does.not.exist"],`
	`99`	`+ ["experts,shared_experts"],`
`100`	`100`	`],`
`101`	`101`	`)`
`102`	`102`	`@pytest.mark.parametrize("compile", [False, True])`