pytorch/torch/csrc/jit/codegen
Ivan Yashchuk 308b26fe4d Add nvFuser support for transpose (#84629)
`torch._refs.t`, `torch._refs.transpose`, `torch._refs.permute` are all should be working now with nvFuser executor. It would also work with graphs processed by AOT Autograd as these functions are registered to the aten->ref mapping via the "register_decomposition" decorator:
07d398fb26/torch/_refs/__init__.py (L3125-L3126)
07d398fb26/torch/_refs/__init__.py (L3143-L3144)
07d398fb26/torch/_refs/__init__.py (L2548-L2549)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/84629
Approved by: https://github.com/ngimel
2022-09-21 12:45:15 +00:00
..
cuda Add nvFuser support for transpose (#84629) 2022-09-21 12:45:15 +00:00
fuser [ROCm] enable jiterator (#77982) 2022-08-15 16:04:09 +00:00
onednn simple c10 implementation for std::call_once (#78051) 2022-06-28 15:47:03 +00:00