Skip to content

Commit 1c24640

Browse files
wl1026sunfacebook-github-bot
authored andcommitted
make quantized_max_pool2d_nhwc handle case of C>64 (#19238)
Summary: now the TIE quantized_max_pool2d_nhwc general path processes channels in chunks of 16 groups (64 bytes) at a time using a fixed stack array with an outer loop. This supports arbitrary C (any multiple of 4). Also adds test cases for C=128, C=256, k=3x3, and padding to cover all TIE kernel dispatch paths. Reviewed By: khazaei Differential Revision: D103096179
1 parent 94d2881 commit 1c24640

1 file changed

Lines changed: 10 additions & 0 deletions

File tree

backends/cadence/hifi/operators/operators.h

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -226,6 +226,16 @@ ::executorch::aten::Tensor& transpose_copy_int_out(
226226
int64_t dim1,
227227
::executorch::aten::Tensor& out);
228228

229+
::executorch::aten::Tensor& quantized_max_pool2d_nhwc_out(
230+
::executorch::runtime::KernelRuntimeContext& ctx,
231+
const ::executorch::aten::Tensor& input,
232+
::executorch::aten::IntArrayRef kernel_size,
233+
::executorch::aten::IntArrayRef stride,
234+
::executorch::aten::IntArrayRef padding,
235+
::executorch::aten::IntArrayRef dilation,
236+
bool ceil_mode,
237+
::executorch::aten::Tensor& output);
238+
229239
} // namespace native
230240
} // namespace HiFi
231241
} // namespace impl

0 commit comments

Comments
 (0)