update assert msg

danielvegamyhre · danielvegamyhre · commit 2c4f70b73172 · 2025-03-28T12:33:26.000-07:00
diff --git a/torch/_inductor/fx_passes/micro_pipeline_tp.py b/torch/_inductor/fx_passes/micro_pipeline_tp.py
@@ -870,7 +870,7 @@ def fuse_matmul_reduce_scatter(reduce_scatter: _ReduceScatterMatch) -> None:
     )
 
     assert is_symm_mem_enabled_for_group(group_name), (
-        f"symmetric memory is not enabled for process group {group_name}, skipping fuse_matmul_reduce_scatter fusion"
+        f"symmetric memory is not enabled for process group {group_name}, this is required for async TP"
     )
 
     # Currently fused_matmul_reduce_scatter doesn't return the matmul result,

Original file line number	Diff line number	Diff line change
`@@ -870,7 +870,7 @@ def fuse_matmul_reduce_scatter(reduce_scatter: _ReduceScatterMatch) -> None:`
`870`	`870`	`)`
`871`	`871`
`872`	`872`	`assert is_symm_mem_enabled_for_group(group_name), (`
`873`		`- f"symmetric memory is not enabled for process group {group_name}, skipping fuse_matmul_reduce_scatter fusion"`
	`873`	`+ f"symmetric memory is not enabled for process 474E group {group_name}, this is required for async TP"`
`874`	`874`	`)`
`875`	`875`
`876`	`876`	`# Currently fused_matmul_reduce_scatter doesn't return the matmul result,`