[ET-VK] Removing un used push constants for conv2d pw. (#10841)

Pull Request resolved: #10814

This change removes unused push constants from conv2d pw op to reduce memory usage.
ghstack-source-id: 283597017
@exported-using-ghexport

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

Author: pytorchbot

backends/vulkan/runtime/graph/ops/glsl/conv2d_pw.glsl

@@ -29,12 +29,8 @@ ${layout_declare_tensor(3, "r", "t_bias", DTYPE, "texture2d")}
 
 layout(push_constant) uniform restrict Block {
   ivec4 out_limits;
-  ivec4 in_sizes;
-  ivec2 kernel_size;
   ivec2 stride;
   ivec2 padding;
-  ivec2 dilation;
-  ivec2 overlay_region;
   int in_group_size;
   int dummy_padding;
   float out_min;

backends/vulkan/runtime/graph/ops/impl/Convolution.cpp

@@ -404,7 +404,27 @@ void add_conv2d_node(
 
   vkapi::ParamsBindList param_buffers;
   std::vector<PushConstantDataInfo> push_constants;
-  if (method == Conv2dMethod::Pointwise || method == Conv2dMethod::Depthwise) {
+  if (method == Conv2dMethod::Pointwise) {
+    const utils::ivec4 kernel_param_stride_pad = {
+        kernel_params.stride[0],
+        kernel_params.stride[1],
+        kernel_params.padding[0],
+        kernel_params.padding[1],
+    };
+
+    struct Conv2dPWParams final {
+      int in_group_size;
+      int dummy_padding;
+      OutputParams out_params;
+    } param{extra_params.in_group_size, 0, out_params};
+
+    push_constants = {
+        graph.logical_limits_pc_of(out),
+        PushConstantDataInfo(
+            &kernel_param_stride_pad, sizeof(kernel_param_stride_pad)),
+        PushConstantDataInfo(&param, sizeof(param)),
+    };
+  } else if (method == Conv2dMethod::Depthwise) {
     const utils::ivec4 kernel_param_size_stride = {
         kernel_params.kernel_size[0],
         kernel_params.kernel_size[1],