Update semi_auto_parallel_sharding_stage_1.py

Xing-lil · web-flow · commit 900f276d2a5a · 2025-05-12T17:57:48.000+08:00
diff --git a/test/auto_parallel/semi_auto_parallel_sharding_stage_1.py b/test/auto_parallel/semi_auto_parallel_sharding_stage_1.py
@@ -175,7 +175,7 @@ def test_sharding_stage_1_overlap_to_static(self):
 
     def test_pure_sharding_multi_mesh_stage_1_with_inplace_master_grad(self):
         def run_sharding_test(enable_inplace_master_grad):
-            os.environ['FLAGS_enable_inplace_master_grad'] = (
+            os.environ['FLAGS_enable_tensor_fusion'] = (
                 '1' if enable_inplace_master_grad else '0'
             )
             paddle.distributed.auto_parallel.set_mesh(self._multi_dim_mesh)

Original file line number	Diff line number	Diff line change
`@@ -175,7 +175,7 @@ def test_sharding_stage_1_overlap_to_static(self):`
`175`	`175`
`176`	`176`	`def test_pure_sharding_multi_mesh_stage_1_with_inplace_master_grad(self):`
`177`	`177`	`def run_sharding_test(enable_inplace_master_grad):`
`178`		`- os.environ['FLAGS_enable_inplace_master_grad'] = (`
	`178`	`+ os.environ['FLAGS_enable_tensor_fusion'] = (`
`179`	`179`	`'1' if enable_inplace_master_grad else '0'`
`180`	`180`	`)`
`181`	`181`	`paddle.distributed.auto_parallel.set_mesh(self._multi_dim_mesh)`