ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-12 17:26:00 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
9f7c1930646ae54e90644a6f869f92b70b2dcdba
composable_kernel/test/batched_gemm_softmax_gemm
History
Anthony Chang 7c788e10ce Add batched attention special kernel instances (#424) 
* sanity check

* add attribution

* add irrgular k tile size for batched attention

* format
2022-09-19 19:20:54 -05:00
..
CMakeLists.txt
Fused attention (#345)
2022-08-13 00:16:14 -05:00
test_batched_gemm_softmax_gemm_fp16.cpp
Add batched attention special kernel instances (#424)
2022-09-19 19:20:54 -05:00
test_batched_gemm_softmax_gemm_util.hpp
Add batched attention special kernel instances (#424)
2022-09-19 19:20:54 -05:00