From 3ed106d4631d888b83e674f0d26598d6dc935361 Mon Sep 17 00:00:00 2001 From: Abhishek Chandra Date: Fri, 7 Feb 2025 13:13:19 -0800 Subject: [PATCH] Convolution 1D optimized axis map (#8312) Summary: Add the kOptimizedAxisMap flag for conv1D. Reviewed By: nathanaelsee, SS-JIA Differential Revision: D67692963 --- backends/vulkan/runtime/graph/ops/impl/Convolution.cpp | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/backends/vulkan/runtime/graph/ops/impl/Convolution.cpp b/backends/vulkan/runtime/graph/ops/impl/Convolution.cpp index 71b7ce80cc0..0515f59e653 100644 --- a/backends/vulkan/runtime/graph/ops/impl/Convolution.cpp +++ b/backends/vulkan/runtime/graph/ops/impl/Convolution.cpp @@ -475,7 +475,12 @@ void add_conv1d_node( const ValueRef out, const bool clamp_out) { ValueRef arg_weight = prepack_standard( - graph, weight, graph.storage_type_of(out), utils::kChannelsPacked); + graph, + weight, + graph.storage_type_of(out), + utils::kChannelsPacked, + false, + utils::kOptimizedAxisMap); ValueRef arg_bias = prepack_biases( graph, bias,