pytorch
9f03444f - Add torch.ops.aten -> torch._refs mapping to TorchRefsMode using decomposition_table (#82657)

Commit
2 years ago
Add torch.ops.aten -> torch._refs mapping to TorchRefsMode using decomposition_table (#82657) ### Description This PR adds the possibility to convert `torch.ops.aten` calls to `torch._refs` and consequently prims under TorchRefsMode. ### Testing New test, `test_aten_overload_to_prims`, in `test/test_prims.py`. Pull Request resolved: https://github.com/pytorch/pytorch/pull/82657 Approved by: https://github.com/jjsjann123, https://github.com/ezyang
Author
Committer
Parents
Loading