Skip to content

Commit da43151

Browse files
committed
Update on "[ET-VK] Adding more pointwise convolution tests."
This diff adds new test cases for pointwise convolution in the Vulkan backend of Executorch. The updated test cases cover different configurations of convolution layers, including kernel size, padding, stride, and groups. Differential Revision: [D75961457](https://our.internmc.facebook.com/intern/diff/D75961457/) [ghstack-poisoned]
2 parents c5cae76 + 5ef3407 commit da43151

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

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

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -211,7 +211,8 @@ ValueRef prepack_weights(
211211
vkapi::ShaderInfo shader =
212212
get_conv2d_shader(graph, *t, /*prepack_weights = */ true, method, vref);
213213

214-
const auto sizes_pc = utils::make_ivec4(original_sizes, /*reverse = */ true);
214+
const auto original_sizes_pc =
215+
utils::make_ivec4(original_sizes, /*reverse = */ true);
215216
graph.prepack_nodes().emplace_back(new PrepackNode(
216217
graph,
217218
shader,
@@ -223,7 +224,7 @@ ValueRef prepack_weights(
223224
// Specialization constants
224225
{SV(t->packed_dim())},
225226
{graph.sizes_pc_of(v),
226-
PushConstantDataInfo(&sizes_pc, sizeof(sizes_pc))}));
227+
PushConstantDataInfo(&original_sizes_pc, sizeof(original_sizes_pc))}));
227228

228229
return v;
229230
}

0 commit comments

Comments
 (0)