Skip to content

Commit 0e9cf7f

Browse files
committed
Update on "[ET-VK] Using push constants for convolution prepack dispatches."
Differential Revision: [D74523774](https://our.internmc.facebook.com/intern/diff/D74523774/) [ghstack-poisoned]
2 parents d4bc9a6 + a95b7af commit 0e9cf7f

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)