pytorch/torch/csrc/jit/codegen
Ivan Yashchuk 4a8382b58e Update caching of tensor arguments for nvFuser's fusion creation (#87860)
Previously nvFuser's fusion definition was cached based on concrete shape and strides of tensor inputs for simplicity and correctness. This PR changes Python's cache to check the number of dimensions, size-1 dimensions, and contiguity information based on given strides and shapes.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/87860
Approved by: https://github.com/kevinstephano, https://github.com/jjsjann123, https://github.com/ngimel
2022-11-02 09:29:20 +00:00
..
cuda Update caching of tensor arguments for nvFuser's fusion creation (#87860) 2022-11-02 09:29:20 +00:00
fuser [BE] Delete TH_DISALLOW_COPY_AND_ASSIGN (#87743) 2022-10-26 03:31:56 +00:00
onednn Fix typos under torch directory (#88172) 2022-11-01 22:58:22 +00:00