Skip to content

Commit 5dfcad1

Browse files
authored
Update Convolution.cpp
1 parent c19fe4a commit 5dfcad1

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

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

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -402,6 +402,7 @@ void add_conv2d_node(
402402
wg_size = {wg_size[0] * wg_size[1] * wg_size[2], 1, 1};
403403
} else if (method == Conv2dMethod::Pointwise) {
404404
wg_size = {wg_size[0] * wg_size[1], wg_size[2], 1};
405+
}
405406

406407
utils::uvec3 local_wg_size;
407408
if (method == Conv2dMethod::Pointwise) {

0 commit comments

Comments
 (0)