frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

Increase max total number of dynamo partitions to 15 (#134153)

Browse Source 
Needed to be able to split some of the aarch64 workflows to 15 shards

Pull Request resolved: https://github.com/pytorch/pytorch/pull/134153
Approved by: https://github.com/seemethere, https://github.com/kit1980, https://github.com/ZainRizvi
This commit is contained in:
Nikita Shulga
2024-08-21 23:10:12 +00:00
committed by PyTorch MergeBot
parent a5ef04a3b8
commit 5f0bd98767
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
benchmarks/dynamo/common.py
View File

@ -3455,7 +3455,7 @@ def parse_args(args=None):
"--total-partitions",
type=int,
default=1,
choices=range(1, 10),
choices=range(1, 16),
help="Total number of partitions we want to divide the benchmark suite into",
)
parser.add_argument(