pytorch
39e8311a - unwrap sizevars passed as args (#96051)

Commit
2 years ago
unwrap sizevars passed as args (#96051) Undoes int arg wrapping that dynamo does Generated line: ``` s1 = arg2_1.item() if isinstance(arg2_1, torch.Tensor) else arg2_1 ``` Fixes #ISSUE_NUMBER Pull Request resolved: https://github.com/pytorch/pytorch/pull/96051 Approved by: https://github.com/ezyang, https://github.com/jansel
Author
Natalia Gimelshein
Committer
Parents
Loading