SIGN IN SIGN UP

[ET-VK] Replacing the use of uvec3 with WorkgroupSize class to reduce memory usage and improve processing speed (#8671)

* [ET-VK] Adding a workgroup class to VecUtils

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

This diff adds a new class called `WorkgroupSize` to the `VecUtils` header file. The `WorkgroupSize` class takes three `uint32_t` values as parameters and stores them in a single `uint32_t` variable using bitwise operations. This class is used in the Vulkan backend to specify the size of a workgroup for a given operation.
ghstack-source-id: 268172661
@exported-using-ghexport

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

* [ET-VK] Adding reserve and append functions to SpecVarList

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

This diff adds two new functions to the SpecVarList class in the Vulkan runtime library. The first function, reserve, allows the user to reserve a certain amount of space in the SpecVarList before adding any elements. The second function, append, allows the user to add a single SpecVar to the SpecVarList. These functions are useful for optimizing memory usage and improving performance in the Vulkan runtime.
ghstack-source-id: 268172659
@exported-using-ghexport

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

* [ET-VK] Replacing the use of uvec3 with WorkgroupSize class to reduce memory usage and improve processing speed

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

This diff replaces the use of `uvec3` with `WorkgroupSize` class to reduce memory usage and improve processing speed in the Vulkan backend of Executorch.
ghstack-source-id: 268172660
@exported-using-ghexport

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

---------

Co-authored-by: Vivek Trivedi <5340687+trivedivivek@users.noreply.github.com>
P
pytorchbot committed
973eb1c4f2f076d382c881d1b133f601ceddd0b4
Parent: 54910bc
Committed by GitHub <noreply@github.com> on 2/25/2025, 3:56:54 PM