pytorch
90161c23 - Add nvfuser support for squeeze (#84117)

Commit
2 years ago
Add nvfuser support for squeeze (#84117) "_refs.squeeze" and "refs.unsqueeze" now work with nvfuser executor tests. Similarly to `_refs.reshape` we need to explicitly save the concrete shape on the trace to pass that info to nvfuser, as it gets lost in translation (https://github.com/pytorch/pytorch/pull/83739#discussion_r950352124). Pull Request resolved: https://github.com/pytorch/pytorch/pull/84117 Approved by: https://github.com/ngimel
Author
Committer
Parents
Loading