saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Re-enable dynamic shapes test in dynamo benchmark (#99816)

Browse source 
Set `torch._dynamo.config.assume_static_by_default = False` for dynamic shapes flag enabled

Fixes #99815

Pull Request resolved: https://github.com/pytorch/pytorch/pull/99816
Approved by: https://github.com/jgong5, https://github.com/ezyang
This commit is contained in:
chuanqiw 2023-04-24 20:34:52 +00:00 committed by PyTorch MergeBot
parent d881b2978c
commit e9e5ffe83e
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
benchmarks/dynamo/common.py
View file

@ -2053,6 +2053,8 @@ def run(runner, args, original_dir=None):
torch._dynamo.config.assume_static_by_default = True
if args.dynamic_shapes:
torch._dynamo.config.dynamic_shapes = True
if not args.dynamic_batch_only:
torch._dynamo.config.assume_static_by_default = False
if args.specialize_int:
torch._dynamo.config.specialize_int = True
if args.ci: