pytorch
006785cd - [dynamo][hf_bigbird] Actually graph break on tensor.unsqueeze_/resize_ (#99986)

Commit
1 year ago
[dynamo][hf_bigbird] Actually graph break on tensor.unsqueeze_/resize_ (#99986) Currently, we return `unimplemented` w/o a graph break on seeing a x.unsqueeze_ when x is input. This essentially means we fall back to the original frame. This PR actually graph breaks so that we can generate the continuation frame for the rest of the function. Instead of graph breaking at LOAD_ATTR, we delay the graph break to the actual CALL_FUNCTION, where its cleaner to graph break. Pull Request resolved: https://github.com/pytorch/pytorch/pull/99986 Approved by: https://github.com/jansel
Author
Committer
Parents
Loading