RFC: Implementation of faster TopK.

Lubo Litchev

unread,

Apr 23, 2024, 2:29:46 PM4/23/24

to iree-discuss

Hey folks,

We have run into some perf issues of the (CPU) topk and I have been experimenting with some implementation.

Here is an RFC for speeding things up...
https://docs.google.com/document/d/163WR31ZAdSGchSVT_vRntcTZ4GqMwUgI0LJybtrS9BI/edit?usp=sharing&resourcekey=0-xf1MqxXiF_xg1y2-EPH7WA

Scott Todd

unread,

Apr 23, 2024, 2:44:00 PM4/23/24

to Lubo Litchev, iree-discuss

I don't have access to that document. Can you make it publicly viewable or post it somewhere like a GitHub issue or gist?

--
You received this message because you are subscribed to the Google Groups "iree-discuss" group.
To unsubscribe from this group and stop receiving emails from it, send an email to iree-discuss...@googlegroups.com.
To view this discussion on the web visit https://groups.google.com/d/msgid/iree-discuss/843b8cbd-7997-49ba-bab2-11a13e56c8fdn%40googlegroups.com.

Hanhan Wang

unread,

Apr 25, 2024, 4:10:45 PM4/25/24

to Scott Todd, Lubo Litchev, iree-discuss

It is available on Github issue: https://github.com/iree-org/iree/issues/17143

To view this discussion on the web visit https://groups.google.com/d/msgid/iree-discuss/CAB%3DqEE6pSoFTqodnNt7sbtZucQVxVEkHfgXtGFrVsJF%2BJXTifg%40mail.gmail.com.

Reply all

Reply to author

Forward