Fixing linux build. (#7580)

Summary:

Fixing build issue in unittest / linux / linux-job

2025-01-09T21:17:15.4021836Z /pytorch/executorch/cmake-out/conv2d_dw_half.glsl:58: error: 'idx_to_ipos_x_wise' : no matching overloaded function found
2025-01-09T21:17:15.4022815Z /pytorch/executorch/cmake-out/conv2d_dw_half.glsl:58: error: 'const' : non-matching or non-convertible constant type for const initializer
2025-01-09T21:17:15.4023491Z 2 errors generated.
2025-01-09T21:17:15.4024083Z /pytorch/executorch/cmake-out/conv2d_dw_output_tile_3x3_half.glsl:70: error: 'idx_to_ipos_x_wise' : no matching overloaded function found
2025-01-09T21:17:15.4025172Z /pytorch/executorch/cmake-out/conv2d_dw_output_tile_3x3_half.glsl:70: error: '=' :  cannot convert from ' const float' to ' temp highp 3-component vector of int'
2025-01-09T21:17:15.4025903Z 2 errors generated.
2025-01-09T21:17:15.4026440Z /pytorch/executorch/cmake-out/conv2d_pw_half.glsl:67: error: 'idx_to_ipos_x_wise' : no matching overloaded function found
2025-01-09T21:17:15.4027412Z /pytorch/executorch/cmake-out/conv2d_pw_half.glsl:67: error: 'const' : non-matching or non-convertible constant type for const initializer

Reviewed By: jorgep31415

Differential Revision: D67997666

Author: trivedivivek

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

@@ -35,7 +35,8 @@ layout(local_size_x_id = 0, local_size_y_id = 1, local_size_z_id = 2) in;
  * output at a single output location.
  */
 void main() {
-  const ivec3 pos = idx_to_ipos_x_wise(gl_GlobalInvocationID.x, out_limits.x, out_limits.y);
+  const ivec3 gid = ivec3(gl_GlobalInvocationID);
+  const ivec3 pos = idx_to_ipos_x_wise(gid.x, out_limits.x, out_limits.y);
 
   if (any(greaterThanEqual(pos, out_limits))) {
     return;

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

@@ -47,7 +47,8 @@ void main() {
   // since work size is calculated by x * ((y + B_Y - 1) / B_Y) * z
   const ivec2 out_limits_xy_scaled = (out_limits.xy + ivec2(BATCH_SIZE_X, BATCH_SIZE_Y) - 1) / ivec2(BATCH_SIZE_X, BATCH_SIZE_Y);
 
-  ivec3 pos = idx_to_ipos_x_wise(gl_GlobalInvocationID.x, out_limits_xy_scaled.x, out_limits_xy_scaled.y);
+  const ivec3 gid = ivec3(gl_GlobalInvocationID);
+  ivec3 pos = idx_to_ipos_x_wise(gid.x, out_limits_xy_scaled.x, out_limits_xy_scaled.y);
 
   // scale pos.xy by batch sizes, because that's the top pixel to be processed
   pos.x *= BATCH_SIZE_X;

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

@@ -44,7 +44,8 @@ void main() {
   const ivec2 out_limits_scaled = (out_limits.xy + TILE_SIZE - 1) / TILE_SIZE;
   const uint shared_mem_stride = gl_WorkGroupSize.x * gl_WorkGroupSize.y * gl_WorkGroupSize.z;
 
-  const ivec3 gpos = idx_to_ipos_x_wise(gl_GlobalInvocationID.x, out_limits_scaled.x, out_limits_scaled.y);
+  const ivec3 gid = ivec3(gl_GlobalInvocationID);
+  const ivec3 gpos = idx_to_ipos_x_wise(gid.x, out_limits_scaled.x, out_limits_scaled.y);
 
   // Output position for TILE_SIZE = 2
   // +--------+--------+

backends/vulkan/runtime/graph/ops/glsl/indexing_utils.h

@@ -223,8 +223,8 @@ ivec3 lpos_to_pos(const ivec3 lpos, const ivec4 axis_map) {
   return pos;
 }
 
-ivec3 idx_to_ipos_x_wise(uint idx, int size_x, int size_y) {
-  const uint div_by_x = idx / size_x;
+ivec3 idx_to_ipos_x_wise(int idx, int size_x, int size_y) {
+  int div_by_x = idx / size_x;
   return ivec3(idx % size_x, div_by_x % size_y, div_by_x / size_y);
 }