Skip to content

Commit 5ef3407

Browse files
committed
Update base for 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]
1 parent d4bc9a6 commit 5ef3407

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)