ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-03 13:11:25 +00:00
Code Issues Packages Projects Releases Wiki Activity
2,623 Commits 734 Branches 37 Tags
4a8b17d1a4e285b5925b78cc5c4cd3bb2fec95e7
Commit Graph

2 Commits

Author SHA1 Message Date
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
carlushuang
b098b71b05 topk_softmax (#1592) 
* topk_softmax

* remove some file

* fix atomix linear_offset

* address various comment, and change sfc get_index api to static(tuple)
 2024-10-26 23:52:49 +08:00