pytorch
6cfaa922 - Handle tensor default func args when inlining (#90575)

Commit
2 years ago
Handle tensor default func args when inlining (#90575) Handle tensor default func/method args when inlining Previously, when inlining a function, its default arguments were only wrapped with VariableTrackers if non-tensor. Now, tensor default args are also handled by adding them to the parent InstructionTranslator as an attribute. - also patches up a missing source in nnmodule call_function, needed to properly guard on a default arg in its methods - adds new 'DefaultsSource' type which guards either a `__defaults__` or `__kwdefaults__` entry on a function Fixes #90361 https://github.com/pytorch/torchdynamo/issues/1968 Pull Request resolved: https://github.com/pytorch/pytorch/pull/90575 Approved by: https://github.com/voznesenskym
Author
Committer
Parents
Loading