[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