You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
[ET-VK] Introduce BufferMetadata GLSL struct to abstract tensor layout
Pull Request resolved: #13595
As title; introduce an consolidated metadata UBO for buffer storage that can be used to abstract tensor indexing operations for buffer-backed tensors.
This new metadata UBO is capable of representing tensors of up to 8 dimensions. This upper limit is hardcoded, but can be increased later on without needing to update callsites since everything is abstracted by the BufferMetadata struct.
Update the following ops to use this new metadata UBO:
* staging shaders (nchw_to_buffer and buffer_to_nchw)
* binary op
@imported-using-ghimport
Differential Revision: [D80800082](https://our.internmc.facebook.com/intern/diff/D80800082/)
ghstack-source-id: 305143836
0 commit comments