Update on "[ET-VK] Replace Uniform buffers with push constants for permute op"

This diff replaces uniform buffers with push constants for permute op in the Vulkan backend of Executorch. The changes include updating the GLSL code to use push constants instead of uniform buffers and updating the C++ code to pass the sizes as push constants to the shader.

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

[ghstack-poisoned]

Author: trivedivivek

backends/vulkan/test/vulkan_compute_api_test.cpp

@@ -1601,6 +1601,7 @@ TEST(VulkanComputeGraphTest, test_simple_shared_objects_with_resize) {
   auto addFn = VK_GET_OP_FN("aten.add.Tensor");
   addFn(graph, {a.value, b.value, kDummyValueRef, c});
 
+  // no new allocations if binary op uses push constants
   expected_vma_allocation_count += 0;
   EXPECT_EQ(get_vma_allocation_count(), expected_vma_allocation_count);
 
@@ -1622,9 +1623,8 @@ TEST(VulkanComputeGraphTest, test_simple_shared_objects_with_resize) {
   auto mulFn = VK_GET_OP_FN("aten.mul.Tensor");
   mulFn(graph, {c, d.value, e});
 
-  // +2: alpha UBO, broadcast UBO for arithmetic shader
-  // +1: t.sizes_ubo() for arithmetic shader output e
-  expected_vma_allocation_count += 3;
+  // no new allocations if binary op uses push constants
+  expected_vma_allocation_count += 0;
   EXPECT_EQ(get_vma_allocation_count(), expected_vma_allocation_count);
 
   IOValueRef out = {};