SIGN IN SIGN UP

[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