Skip to content

Commit d550075

Browse files
committed
Update on "[ET-VK] Minor dispatch improvement to conv2d dw op to improve performance."
Differential Revision: [D76242234](https://our.internmc.facebook.com/intern/diff/D76242234/) [ghstack-poisoned]
2 parents dd8836b + b77a025 commit d550075

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -406,7 +406,7 @@ void add_conv2d_node(
406406

407407
utils::uvec3 wg_size = create_conv2d_global_wg_size(
408408
graph, method, out, weight_data, stride_equals_dilation);
409-
409+
410410
utils::uvec3 local_wg_size;
411411
if (method == Conv2dMethod::Depthwise || method == Conv2dMethod::Pointwise) {
412412
wg_size = {wg_size[0] * wg_size[1], wg_size[2], 1};

backends/vulkan/test/op_tests/cases.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -504,7 +504,7 @@ def get_conv_inputs():
504504
transposed=False,
505505
output_padding=[0, 0],
506506
groups=XS,
507-
),
507+
),
508508
Test(
509509
self=(1, XS, S, S1),
510510
weight=(XS, 1, 3, 3),

0 commit comments

Comments
 (0)