Introduce `virtual_transpose()` to `vTensor` for no copy transposition (#5353)
Summary: Pull Request resolved: https://github.com/pytorch/executorch/pull/5353 ## Context With `axis_map` integrated into matrix multiplication, we can now test no-copy transposes for texture backed tensors. Transposing a tensor can be done without modifying the storage by swapping elements in the tensor's `axis_map`, and also updating the layout of the tensor if the packed dimension was one of the dims that were transposed. ghstack-source-id: 242802318 exported-using-ghexport Reviewed By: jorgep31415 Differential Revision: D62652009 fbshipit-source-id: ae1c81a61a92e69aaa689418d7e62a81d8e09a12
S
Stephen Jia committed
26375cce7860556b4b2941c48db43a40209d6e6d
Parent: c252553
Committed by Facebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
on 9/16/2024, 6:54:29 PM