Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Update on "[ET-VK] Adding UniformData struct in vTensor class to stor…
…e uniform data, which will be stored using shared ptr and can be shared with push constants."

This diff adds a new struct called `UniformData` in the `vTensor` class to store uniform data, which can be shared with push constants. The `UniformData` struct contains the sizes, strides, and logical limits of the tensor, as well as the number of elements in the tensor.

Diff adds `Attribute` enum to Tensor class to enumerate attributes supplied to dispatch and `UniformData` class to store tensor data supplied as uniforms to op shaders.

The diff also adds write_attribute function to UniformData class to write attribute data to a given memory.

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

[ghstack-poisoned]
  • Loading branch information
trviv committed Dec 6, 2024
commit b7e994756433b46ef5c959651bb687e2b2d9a66a

This merge commit was added into this branch cleanly.

There are no new changes to show, but you can still view the diff.