pytorch/torch/_dynamo/variables
Joel Schlosser dc7bf1a7ea General reversible binary op support (e.g. __add__ / __radd__) in dynamo (#93271)
Generic support for reversible binary op pairs (e.g. `__add__` / `__radd__`) in dynamo.
Adds logic to flip args and try the reverse op when the forward op is unsupported.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/93271
Approved by: https://github.com/voznesenskym, https://github.com/jansel, https://github.com/ezyang
2023-02-03 19:28:35 +00:00
..
__init__.py Properly guard all numpy usage within dynamo and remove UnspecializedNumpyVariable (#90795) 2023-01-06 22:36:38 +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 [inductor] fix crash issue when input is a view tensor (#90150) 2023-02-03 04:54:14 +00:00
builtin.py General reversible binary op support (e.g. __add__ / __radd__) in dynamo (#93271) 2023-02-03 19:28:35 +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 General reversible binary op support (e.g. __add__ / __radd__) in dynamo (#93271) 2023-02-03 19:28:35 +00:00
misc.py Delete dynamo_import and inductor_import (#93851) 2023-02-02 01:51:29 +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 [dynamo] Handle general tensor attributes with a getattr proxy node (#91840) 2023-02-01 22:34:03 +00:00
torch.py [SDPA] Wire up FlashAttention's backward (#92917) 2023-02-02 04:02:30 +00:00
user_defined.py [Dynamo] UserDefinedClassVariable supports python type (#93310) 2023-01-31 17:41:51 +00:00