Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
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: #87072 Approved by: https://github.com/ngimel
- Loading branch information