[ET-VK] Enforce GPU buffer limit when partitioning (#6856)
Pull Request resolved: https://github.com/pytorch/executorch/pull/6829 ## Context In Vulkan, there is a limit on the number of elements a GPU buffer can have. If a GPU buffer exceeds this limit, then the API will either produce an error or undefined behaviour will ensue. ## Changes Along with `texture_limits`, introduce a configurable `buffer_limit` entry in the partitioner configuration. ghstack-source-id: 253568943 Differential Revision: [D65899828](https://our.internmc.facebook.com/intern/diff/D65899828/) Co-authored-by: Stephen Jia <ssjia@meta.com>
P
pytorchbot committed
0f2995f7549081f3e3192297f73d037349d32f63
Parent: 07c4d0e
Committed by GitHub <noreply@github.com>
on 11/14/2024, 6:18:26 PM