pytorch/torch/_dynamo/variables
Joel Schlosser f954498edf Dynamo: Fix to unpack ConstantVariable in call_range() (#94202)
Fixes the `pyhpc_turbulent_kinetic_energy` model in torchbench.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/94202
Approved by: https://github.com/ezyang, https://github.com/voznesenskym
2023-02-07 15:12:00 +00:00
..
__init__.py [Dynamo] Put torch.cuda.stream into Dynamo FX graph (#93808) 2023-02-05 04:52:43 +00:00
base.py Cache guards once per variable tracker, rather than re-propagating them repeatedly (#89827) 2022-12-02 01:45:05 +00:00
builder.py [Dynamo] Put torch.cuda.stream into Dynamo FX graph (#93808) 2023-02-05 04:52:43 +00:00
builtin.py Dynamo: Fix to unpack ConstantVariable in call_range() (#94202) 2023-02-07 15:12:00 +00:00
constant.py General reversible binary op support (e.g. __add__ / __radd__) in dynamo (#93271) 2023-02-03 19:28:35 +00:00
dicts.py [Dynamo] Support context wrapping(e.g, torch.no_grad) on nested functions w/o closure (#92922) 2023-01-26 04:23:35 +00:00
functions.py Check for none for NNModuleVariable.__module__ (#93326) 2023-02-02 09:41:41 +00:00
lists.py Fix bug in generic_list_compare (#94156) 2023-02-06 19:50:04 +00:00
misc.py [Dynamo] Put torch.cuda.stream into Dynamo FX graph (#93808) 2023-02-05 04:52:43 +00:00
nn_module.py [Dynamo] Fix bug if module calls module with static forward function (#93299) 2023-01-31 06:16:33 +00:00
tensor.py dont graph break on list[SymInt] comparisons (#94054) 2023-02-05 04:47:12 +00:00
torch.py [Dynamo] Support torch.Tensor.fn as TorchVariable, not UserDefinedObjectVariable, preventing graph break (#93243) 2023-02-07 09:26:50 +00:00
user_defined.py [Dynamo] UserDefinedClassVariable supports python type (#93310) 2023-01-31 17:41:51 +00:00