pytorch
ff2569bc - Intercept aten._reshape_alias for nvFuser (#87072)

Commit
2 years ago
Intercept aten._reshape_alias for nvFuser (#87072) This would help forming larger fusion groups. If this won't end up executed by nvFuser then eager mode implementation would call into `.reshape`: https://github.com/pytorch/pytorch/blob/37e9e89afbc3554258545a026fab4cd9e1a4b85d/torch/_prims/nvfuser_prims.py#L552-L553 cc @kevinstephano @jjsjann123 Pull Request resolved: https://github.com/pytorch/pytorch/pull/87072 Approved by: https://github.com/ngimel
Author
Committer
Parents
Loading