[ET-VK] Add int and bool tensor support for many operators (pytorch#15829)

Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at
bottom):
* __->__ pytorch#15829
* pytorch#15796
* pytorch#15795
* pytorch#15794
* pytorch#15793

Title says it all!

Adds `int32` and `uint8` shader variants to a bunch of operators that
don't currently have variants for these dtypes, but should.

This should prevent folks from running into dtype crashes at runtime
when using the Vulkan delegate.

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

Co-authored-by: ssjia <[email protected]>

Author: SS-JIA

backends/vulkan/runtime/graph/ops/glsl/clone.yaml

@@ -7,5 +7,7 @@ clone:
     DTYPE:
       - VALUE: half
       - VALUE: float
+      - VALUE: int32
+      - VALUE: uint8
   shader_variants:
     - NAME: clone

backends/vulkan/runtime/graph/ops/glsl/concat_buffer.yaml

@@ -6,6 +6,7 @@ concat_buffer:
     DTYPE:
       - VALUE: half
       - VALUE: float
+      - VALUE: int32
   shader_variants:
     - NAME: concat_1_buffer
       NUM_INPUTS: 1

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

@@ -113,8 +113,6 @@ void main() {
 
   VEC4_T out_texel = imageLoad(t_out, out_pos);
 
-  VEC4_T test_texel = VEC4_T(-1.0);
-
   for (int comp = 0; comp < 4; ++comp) {
     ivec4 out_tidx = out_read_start_tidx;
     out_tidx[out_packed_dim] += comp;
@@ -124,7 +122,6 @@ void main() {
     // of the previous input batch; if so, then don't overwrite this texel
     // element
     if (out_tidx[concat_dim] < concat_offset) {
-      test_texel[comp] = -5.0;
       continue;
     }
 
@@ -164,7 +161,6 @@ void main() {
             inp${i}_packed_dim);
 
         out_texel[comp] = texelFetch(t_inp${i}, in_posi.xyz, 0)[in_posi.w];
-        test_texel[comp] = out_texel[comp];
         continue;
       }
       else {

backends/vulkan/runtime/graph/ops/glsl/concat_texture.yaml

@@ -6,6 +6,7 @@ concat_texture:
     DTYPE:
       - VALUE: half
       - VALUE: float
+      - VALUE: int32
   shader_variants:
     - NAME: concat_1_texture3d
       NUM_INPUTS: 1

backends/vulkan/runtime/graph/ops/glsl/expand_buffer.yaml

@@ -6,5 +6,6 @@ expand_buffer:
       - VALUE: half
       - VALUE: float
       - VALUE: int32
+      - VALUE: uint8
   shader_variants:
     - NAME: expand_buffer

backends/vulkan/runtime/graph/ops/glsl/full.yaml

@@ -15,5 +15,6 @@ full:
       - VALUE: half
       - VALUE: float
       - VALUE: int32
+      - VALUE: uint8
   shader_variants:
     - NAME: full

backends/vulkan/runtime/graph/ops/glsl/gather_buffer.yaml

@@ -12,5 +12,7 @@ gather_buffer:
     DTYPE:
       - VALUE: half
       - VALUE: float
+      - VALUE: int32
+      - VALUE: uint8
   shader_variants:
     - NAME: gather_buffer

backends/vulkan/runtime/graph/ops/glsl/gather_texture.yaml

@@ -11,5 +11,7 @@ gather_texture:
     DTYPE:
       - VALUE: half
       - VALUE: float
+      - VALUE: int32
+      - VALUE: uint8
   shader_variants:
     - NAME: gather_texture3d

backends/vulkan/runtime/graph/ops/glsl/index_select.yaml

@@ -8,5 +8,6 @@ index_select:
       - VALUE: half
       - VALUE: float
       - VALUE: int32
+      - VALUE: uint8
   shader_variants:
     - NAME: index_select

backends/vulkan/runtime/graph/ops/glsl/index_select_channel.yaml

@@ -8,5 +8,6 @@ index_select_channel:
       - VALUE: half
       - VALUE: float
       - VALUE: int32
+      - VALUE: uint8
   shader_variants:
     - NAME: index_select_channel