OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

--dynamic-ci-skips now works for all backends (#93369)

Browse Source 
Signed-off-by: Edward Z. Yang <ezyang@meta.com>

Pull Request resolved: https://github.com/pytorch/pytorch/pull/93369
Approved by: https://github.com/albanD
This commit is contained in:
Edward Z. Yang 2023-01-31 10:33:05 -05:00 committed by PyTorch MergeBot
parent 4d504a9ce8
commit 811e95a15e
1 changed files with 0 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
benchmarks/dynamo/common.py
View File

@ -1918,11 +1918,6 @@ def run(runner, args, original_dir=None):
if args.dynamic_ci_skips_only:
args.dynamic_shapes = True
args.ci = True
# We only have a CI skip list for aot_eager right now. When inductor
# comes online, add that skip list too.
assert (
args.backend == "aot_eager"
), "--dynamic-ci-skips only works with aot_eager backend at the moment"
if args.dynamic_shapes:
torch._dynamo.config.dynamic_shapes = True
torch._functorch.config.use_dynamic_shapes = True