pytorch/torch/_dynamo/variables
Jason Ansel 4034873a31 [dynamo] Optimize builtin handling (#122035)
Improves `benchmarks/dynamo/microbenchmarks/dynamo_microbenchmarks.py`
from 7.3s to 6.7s.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/122035
Approved by: https://github.com/Skylion007
ghstack dependencies: #122032, #122033, #122034
2024-03-18 18:08:06 +00:00
..
__init__.py Graph break on step closure in optimizer (#121777) 2024-03-14 03:18:23 +00:00
base.py [dynamo] Optimize VariableTracker.__post_init__ (#122034) 2024-03-18 18:08:06 +00:00
builder.py [dynamo] Optimize VariableTracker.__post_init__ (#122034) 2024-03-18 18:08:06 +00:00
builtin.py [dynamo] Optimize builtin handling (#122035) 2024-03-18 18:08:06 +00:00
constant.py [dynamo] Optimize builtin handling (#122035) 2024-03-18 18:08:06 +00:00
ctx_manager.py [dynamo] Support _unsafe_set_version_counter (#121086) 2024-03-11 05:14:42 +00:00
dicts.py [dynamo] Refactor reconstruct() not to return anything (#120150) 2024-02-17 17:13:41 +00:00
distributed.py [dynamo] Tweak naming for module hook bw_state (#121609) 2024-03-12 16:27:56 +00:00
functions.py [dynamo][guards] Use lazy variable tracker for func defaults (#121388) 2024-03-12 22:48:48 +00:00
higher_order_ops.py [Compiled Autograd] Introduce BackwardState capture (#120382) 2024-02-28 20:36:47 +00:00
iter.py Unify MYPYINDUCTOR and MYPY (#118432) 2024-01-27 17:23:20 +00:00
lazy.py Unify MYPYINDUCTOR and MYPY (#118432) 2024-01-27 17:23:20 +00:00
lists.py [dynamo] Refactor reconstruct() not to return anything (#120150) 2024-02-17 17:13:41 +00:00
misc.py [dynamo] Optimize builtin handling (#122035) 2024-03-18 18:08:06 +00:00
nn_module.py [Dynamo] Support lazy module with namedtuple/dict input (#119972) 2024-02-15 23:18:18 +00:00
optimizer.py Graph break on step closure in optimizer (#121777) 2024-03-14 03:18:23 +00:00
sdpa.py [dynamo] Refactor reconstruct() not to return anything (#120150) 2024-02-17 17:13:41 +00:00
tensor.py [dynamo] Optimize VariableTracker.__post_init__ (#122034) 2024-03-18 18:08:06 +00:00
torch_function.py Allow custom attributes for torch function subclasses (#121693) 2024-03-13 17:01:57 +00:00
torch.py [dynamo] Optimize VariableTracker.__post_init__ (#122034) 2024-03-18 18:08:06 +00:00
user_defined.py Improve Dynamo support for torch function and class methods in general (#121365) 2024-03-08 20:03:49 +00:00