ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-04 05:31:24 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
b098b71b05e4c06310f2e74056282a796f3cfd13
composable_kernel/include/ck_tile/core/numeric
History
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
..
bfloat16.hpp
[CK_TILE] float -> bf16 inline asm rtn (#1482)
2024-08-30 15:38:09 +08:00
float8.hpp
Replace the using of __expf by __ocml_exp_f32 to work-around the test_softmax_rank4 failure (#1394)
2024-07-17 09:15:05 -07:00
half.hpp
Replace the using of __expf by __ocml_exp_f32 to work-around the test_softmax_rank4 failure (#1394)
2024-07-17 09:15:05 -07:00
integer.hpp
introducing ck_tile! (#1216)
2024-04-15 19:27:12 -05:00
integral_constant.hpp
remove operator-deref (#1291)
2024-05-15 08:06:50 -07:00
math.hpp
topk_softmax (#1592)
2024-10-26 23:52:49 +08:00
null_type.hpp
layernorm2d forward (#1339)
2024-06-24 08:45:52 +08:00
numeric.hpp
introducing ck_tile! (#1216)
2024-04-15 19:27:12 -05:00
type_convert.hpp
introducing ck_tile! (#1216)
2024-04-15 19:27:12 -05:00
vector_type.hpp
[CK_TILE] FA bwd kernels optimization (#1397)
2024-08-16 13:40:10 -07:00