File tree 2 files changed +4
-4
lines changed
backends/vulkan/runtime/graph/ops 2 files changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -9,8 +9,8 @@ conv2d_pw:
9
9
OPERATOR : X
10
10
NDIM : 3
11
11
DTYPE : float
12
- TILE_SIZE_X : 2
13
- TILE_SIZE_Y : 2
12
+ TILE_SIZE_X : 1
13
+ TILE_SIZE_Y : 4
14
14
generate_variant_forall :
15
15
DTYPE :
16
16
- VALUE : half
Original file line number Diff line number Diff line change @@ -305,8 +305,8 @@ utils::uvec3 create_conv2d_global_wg_size(
305
305
if (method == Conv2dMethod::Pointwise) {
306
306
const utils::uvec3 image_extents = graph.logical_limits_of (out);
307
307
return {
308
- utils::div_up (image_extents[0u ], 2u ),
309
- utils::div_up (image_extents[1u ], 2u ),
308
+ utils::div_up (image_extents[0u ], 1u ),
309
+ utils::div_up (image_extents[1u ], 4u ),
310
310
image_extents[2u ]};
311
311
} else if (method == Conv2dMethod::Depthwise && stride_equals_dilation) {
312
312
const utils::uvec3 image_extents = graph.create_global_wg_size (out);
You can’t perform that action at this time.
0 commit comments