Update (base update)

ruisizhang123 · ruisizhang123 · commit c4e94be0572a · 2025-09-05T23:33:55.000-07:00
[ghstack-poisoned]
diff --git a/autoparallel/autobucketing_util/bucket_plan.py b/autoparallel/autobucketing_util/bucket_plan.py
@@ -177,7 +177,6 @@ def get_simplefsdp_auto_plan(
             bucketed_comm_memory = (
                 2 * comm_size_inp
                 + 2 * comm_size_out
-                + heuristic_info["accumulated_gradient_size"]
                 + heuristic_info["this_step_rs_comm_inp_size"] * 2
                 + heuristic_info["last_step_rs_comm_size"]
             )

Original file line number	Diff line number	Diff line change
`@@ -177,7 +177,6 @@ def get_simplefsdp_auto_plan(`
`177`	`177`	`bucketed_comm_memory = (`
`178`	`178`	`2 * comm_size_inp`
`179`	`179`	`+ 2 * comm_size_out`
`180`		`- + heuristic_info["accumulated_gradient_size"]`
`181`	`180`	`+ heuristic_info["this_step_rs_comm_inp_size"] * 2`
`182`	`181`	`+ heuristic_info["last_step_rs_comm_size"]`
`183`	`182`	`)`