Implement prepack nodes (#6352)
Summary: Pull Request resolved: https://github.com/pytorch/executorch/pull/6352 ## Context This diff implements the idea described in the previous diff in this stack. During export, `et_vk.prepack` nodes will be inserted to convert constant tensors to GPU tensor objects. This makes it so that Vulkan operators will not have to account for the possibility that their arguments can potentially be constant tensor data instead of an actual tensor object. ghstack-source-id: 248867364 Reviewed By: jorgep31415 Differential Revision: D64603666 fbshipit-source-id: d5b63767423891eac77a7a1949d793e348e60b83
S
Stephen Jia committed
339bb74bad2ca32bf043181688cff74100c65e5d
Parent: 75c17c0
Committed by Facebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
on 10/18/2024, 11:05:23 PM