SIGN IN SIGN UP

[ET-VK] Using a single GPU buffer for all tensor uniforms.

Pull Request resolved: https://github.com/pytorch/executorch/pull/7015

This diff changes Tensor class to store all uniforms in a single uniform buffer.

Entities stored in uniforms ie. size, stride, numel and logical limits are now stored in a single buffer and their offsets are stored as unsigned ints in Tensor class.

Other changes includes:
Adding a new ctor for ParamsBuffer class to allow allocation with size without data ptr.

Adding an offset input to Buffer::data function.

Adding an offset parameter to BufferBindInfo ctor, so additional offset can be supplied when binding a buffer.
ghstack-source-id: 256728325
@exported-using-ghexport

Differential Revision: [D65841750](https://our.internmc.facebook.com/intern/diff/D65841750/)

Co-authored-by: Vivek Trivedi <5340687+trivedivivek@users.noreply.github.com>
P
pytorchbot committed
4e039d081fd3b75376d9c41efe8cf89656b4f055
Parent: a9565aa
Committed by GitHub <noreply@github.com> on 12/5/2024, 8:46:37 PM