Skip to content

Commit

Permalink
Update test_auto_parallel_partitioner.py, test=document_fix (#61982)
Browse files Browse the repository at this point in the history
  • Loading branch information
co63oc committed Feb 23, 2024
1 parent c4dbcc8 commit 52498e2
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions test/legacy_test/test_auto_parallel_partitioner.py
Original file line number Diff line number Diff line change
Expand Up @@ -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(
Expand Down Expand Up @@ -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(
Expand Down Expand Up @@ -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]]

Expand Down Expand Up @@ -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]]

Expand Down Expand Up @@ -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],
Expand Down

0 comments on commit 52498e2

Please sign in to comment.