ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-01 20:21:23 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
e135dd518d19a36466ce7c61bb9d3203ec18c8af
composable_kernel/include/ck_tile/ops/topk_softmax/pipeline
History
Sami Remes 515e283091 [CK_TILE] Top-K with Sigmoid kernel (#3062) 
* Add sigmoid option to topk_softmax

* fix formatting

* add to changelog

* Apply suggestions from code review

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

* Use else if

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

---------

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
Co-authored-by: Adam Osewski <19374865+aosewski@users.noreply.github.com>
2025-10-28 10:54:06 -07:00
..
topk_softmax_warp_per_row_pipeline.hpp
[CK_TILE] Top-K with Sigmoid kernel (#3062)
2025-10-28 10:54:06 -07:00
topk_softmax_warp_per_row_policy.hpp
topk_softmax (#1592)
2024-10-26 23:52:49 +08:00
topk_softmax_warp_per_row_problem.hpp
[CK_TILE] Top-K with Sigmoid kernel (#3062)
2025-10-28 10:54:06 -07:00