Skip to content

fix: Enable asm requantized max pooling with differing qinfo#1285

Merged
morgolock merged 1 commit intomainfrom
pr/enable_max_pool_asm_diff_qinfo
Apr 30, 2026
Merged

fix: Enable asm requantized max pooling with differing qinfo#1285
morgolock merged 1 commit intomainfrom
pr/enable_max_pool_asm_diff_qinfo

Conversation

@morgolock
Copy link
Copy Markdown
Contributor

Restrict differing src/dst quantization-info support in CpuPool2dAssemblyWrapperKernel to MAX pooling, while keeping AVG pooling on the generic fallback path.

Fix the quantized multiplier validation check and wire the requantization shifts correctly for the asm pooling path so requantized MAX pooling validates and executes correctly.

Add NEON validate coverage for padded NHWC QASYMM8 MAX with differing quantization info.

Resolves MLCE-1821

Change-Id: I33c99d0d4ea1bf57ed28d0750422403e60e2a276

@morgolock morgolock requested a review from gunes-arm April 27, 2026 22:05
Copy link
Copy Markdown
Contributor

@gunes-arm gunes-arm left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

fix: enable asm requantized max pooling with differing qinfo
-->
fix: Enable asm requantized max pooling with differing qinfo

Comment thread src/cpu/kernels/internal/CpuPool2dAssemblyWrapperKernel.cpp
Comment thread src/cpu/kernels/internal/CpuPool2dAssemblyWrapperKernel.cpp Outdated
Comment thread tests/validation/NEON/PoolingLayer.cpp Outdated
Comment thread tests/validation/NEON/PoolingLayer.cpp
@morgolock morgolock force-pushed the pr/enable_max_pool_asm_diff_qinfo branch from e640361 to 447d3ad Compare April 28, 2026 12:00
@morgolock morgolock requested a review from gunes-arm April 28, 2026 12:03
@gunes-arm gunes-arm changed the title fix: enable asm requantized max pooling with differing qinfo fix: Enable asm requantized max pooling with differing qinfo Apr 29, 2026
Comment thread tests/validation/NEON/PoolingLayer.cpp Outdated
Comment thread tests/validation/NEON/PoolingLayer.cpp
@gunes-arm gunes-arm self-requested a review April 29, 2026 10:32
Copy link
Copy Markdown
Contributor

@gunes-arm gunes-arm left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Accidental approval last time

@morgolock morgolock force-pushed the pr/enable_max_pool_asm_diff_qinfo branch from 447d3ad to 1ff93da Compare April 30, 2026 08:42
@morgolock morgolock requested a review from gunes-arm April 30, 2026 08:42
Restrict differing src/dst quantization-info support in CpuPool2dAssemblyWrapperKernel to MAX pooling, while keeping AVG pooling on the generic fallback path.

Fix the quantized multiplier validation check and wire the requantization shifts correctly for the asm pooling path so requantized MAX pooling validates and executes correctly.

Reject padded QASYMM8_SIGNED AVG pooling in the asm wrapper for same-qinfo configurations, matching the existing QASYMM8 policy.

Add coverage for QASYMM8_SIGNED padded MAX pooling on the asm path for same qinfo and zero-offset differing qinfo.
Add validate coverage for padded NHWC QASYMM8 MAX with differing quantization info.

Resolves MLCE-1821

Signed-off-by: Pablo Marquez Tello <pablo.tello@arm.com>
Change-Id: I33c99d0d4ea1bf57ed28d0750422403e60e2a276
@morgolock morgolock force-pushed the pr/enable_max_pool_asm_diff_qinfo branch from 1ff93da to c94033f Compare April 30, 2026 09:18
@morgolock morgolock merged commit d619e50 into main Apr 30, 2026
2 checks passed
@morgolock morgolock deleted the pr/enable_max_pool_asm_diff_qinfo branch April 30, 2026 10:40
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants