Update base for Update on "[ET-VK] Adding convenience functions in Compute graph to get PushConstantDataInfo for various attributes of a tensor."

This diff adds convenience functions in the Compute graph to get PushConstantDataInfo for various attributes of a tensor.

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

[ghstack-poisoned]

Author: trivedivivek

backends/vulkan/runtime/graph/ops/DispatchNode.cpp

@@ -75,19 +75,21 @@ void DispatchNode::encode(ComputeGraph* graph) {
 
   bind_params_to_descriptor_set(params_, descriptor_set, idx);
 
-  uint8_t push_constants_data[128];
+  std::array<uint8_t, kMaxPushConstantSize> push_constants_data;
   uint32_t push_constants_offset = 0;
 
   for (const auto& push_constant : push_constants_) {
-    push_constants_offset +=
-        push_constant.write(push_constants_data, push_constants_offset, 128);
+    push_constants_offset += push_constant.write(
+        push_constants_data.data(),
+        push_constants_offset,
+        kMaxPushConstantSize);
   }
   context->register_shader_dispatch(
       descriptor_set,
       pipeline_barrier,
       shader_,
       global_workgroup_size_,
-      push_constants_data,
+      push_constants_data.data(),
       push_constants_offset);
 
   context->report_shader_dispatch_end();

backends/vulkan/runtime/graph/ops/DispatchNode.h

@@ -18,6 +18,7 @@ namespace vkcompute {
 
 class ComputeGraph;
 
+constexpr uint32_t kMaxPushConstantSize = 128;
 /*
  * Represents a push constant data entry
  * Which is either shared pointer to a tensor's uniform data with an attribute