diff --git a/backends/vulkan/runtime/graph/ops/glsl/conv2d_pw_tiled.yaml b/backends/vulkan/runtime/graph/ops/glsl/conv2d_pw_tiled.yaml index 037dfc35c89..82a6dcd24e2 100644 --- a/backends/vulkan/runtime/graph/ops/glsl/conv2d_pw_tiled.yaml +++ b/backends/vulkan/runtime/graph/ops/glsl/conv2d_pw_tiled.yaml @@ -11,7 +11,7 @@ conv2d_pw_tiled: TILE_M4: 1 TILE_K4: 1 TILE_N4: 1 - TILE_M: 4 + TILE_M: 2 generate_variant_forall: DTYPE: - VALUE: float diff --git a/backends/vulkan/runtime/graph/ops/impl/Conv2dPW.cpp b/backends/vulkan/runtime/graph/ops/impl/Conv2dPW.cpp index 7c5f9f0bc0a..3c227448cf2 100644 --- a/backends/vulkan/runtime/graph/ops/impl/Conv2dPW.cpp +++ b/backends/vulkan/runtime/graph/ops/impl/Conv2dPW.cpp @@ -79,8 +79,8 @@ GlobalWorkGrid pick_conv2d_pw_tiled_gwg( uint32_t C_out = graph->size_at(-3, out); uint32_t M = H * W; uint32_t N4 = utils::div_up_4(C_out); - // TILE_N4=1, TILE_M=4 - return GlobalWorkGrid({N4, utils::div_up(M, 4u), 1u}, kTiledWorkGrid); + // TILE_N4=1, TILE_M=2. Must match conv2d_pw_tiled.yaml. + return GlobalWorkGrid({N4, utils::div_up(M, 2u), 1u}, kTiledWorkGrid); } //