From a6619333354b165c6ff9ee08d1e4275d7ffa10f7 Mon Sep 17 00:00:00 2001 From: Nathanael See Date: Fri, 21 Feb 2025 14:03:39 -0800 Subject: [PATCH] Convolution 1D optimized axis map (#8628) Summary: Resubmit D67692963 (https://github.com/pytorch/executorch/pull/8312) Add the kOptimizedAxisMap flag for conv1D. Reviewed By: jorgep31415 Differential Revision: D70009298 --- 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..18599ed4ba6 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, + /* passthrough = */ false, + utils::kOptimizedAxisMap); ValueRef arg_bias = prepack_biases( graph, bias,