Skip to content

NEON implementation for TopK/TopKV #1227

@allnes

Description

@allnes

Hi,

We rely on it in Arm CPU pipelines. I see TopK only via CPPTopKV (arm_compute/runtime/CPP/functions/CPPTopKV.*, arm_compute/core/CPP/kernels/CPPTopKVKernel.*) and no NEON path in the current main branch (checked 2025-12-08).

Question: is a NEON (or other accelerated) TopK/TopKV already available in a private/dev branch or planned on your roadmap? If not, could you consider adding it?

Use case: detection/NMS post-processing; TopK is a visible CPU bottleneck.

Metadata

Metadata

Assignees

No one assigned

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions