From 52498e2a302cd6f09c126219c04c4879182c26e2 Mon Sep 17 00:00:00 2001 From: co63oc Date: Fri, 23 Feb 2024 12:10:12 +0800 Subject: [PATCH] Update test_auto_parallel_partitioner.py, test=document_fix (#61982) --- test/legacy_test/test_auto_parallel_partitioner.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/test/legacy_test/test_auto_parallel_partitioner.py b/test/legacy_test/test_auto_parallel_partitioner.py index 1a0d70c232b36..da402d93ce58e 100644 --- a/test/legacy_test/test_auto_parallel_partitioner.py +++ b/test/legacy_test/test_auto_parallel_partitioner.py @@ -480,7 +480,7 @@ def test_mlp_mp(self): self.assertTrue( distributed_attr_check_for_program(dist_main_prog, dist_context) ) - # check distribured attr for dist op + # check distributed attr for dist op serial_op_idx = [1, 4] dist_op_idx = [[1, 2], [4, 5]] self.assertTrue( @@ -573,7 +573,7 @@ def test_mlp_dp_mp(self): self.assertTrue( distributed_attr_check_for_program(dist_main_prog, dist_context) ) - # check distribured attr for dist op + # check distributed attr for dist op serial_op_idx = [1, 4] dist_op_idx = [[1, 2], [4, 5]] self.assertTrue( @@ -869,7 +869,7 @@ def test_attn_mp(self): self.assertTrue( distributed_attr_check_for_program(dist_main_prog, dist_context) ) - # check distribured attr for dist op + # check distributed attr for dist op serial_op_idx = [0, 4, 6, 18] dist_op_idx = [[0, 1], [4, 5], [6, 7], [18, 19]] @@ -976,7 +976,7 @@ def test_attn_dp_mp(self): self.assertTrue( distributed_attr_check_for_program(dist_main_prog, dist_context) ) - # check distribured attr for dist op + # check distributed attr for dist op serial_op_idx = [0, 4, 6, 18] dist_op_idx = [[0, 1], [4, 5], [6, 7], [18, 19]] @@ -1364,7 +1364,7 @@ def test_decoder_dp_mp(self): self.assertTrue( distributed_attr_check_for_program(dist_main_prog, dist_context) ) - # check distribured attr + # check distributed attr serial_op_idx = [0, 5, 9, 11, 24, 29, 32] dist_op_idx = [ [2, 3],