PoYen, Chen 1c070380fa Merge branch 'feature/cond-add-splitkv' into feature/fmha-fwd-appendkv

2024-07-08 07:13:34 +00:00

..

fmha_bwd_kernel.hpp

CK Tile FA Training kernels (#1286)

2024-06-04 13:12:45 -05:00

fmha_bwd_tile_partitioner.hpp

CK Tile FA Training kernels (#1286)

2024-06-04 13:12:45 -05:00

fmha_fwd_appendkv_kernel.hpp

Fix vnew append errro

2024-06-26 17:00:07 +00:00

fmha_fwd_appendkv_tile_partitioner.hpp

Fix wrong K values after appending

2024-06-25 10:12:13 +00:00

fmha_fwd_kernel.hpp

Fix in dropout lambda to avoid the compiling issue on some docker/compiler envs (#1350)

2024-06-20 11:36:42 +08:00

fmha_fwd_splitkv_combine_kernel.hpp

[CK_TILE] fmha forward split-kv + combine kernels (#1338)

2024-06-26 17:41:15 +08:00

fmha_fwd_splitkv_combine_tile_partitioner.hpp

[CK_TILE] fmha forward split-kv + combine kernels (#1338)

2024-06-26 17:41:15 +08:00

fmha_fwd_splitkv_kernel.hpp

[CK_TILE] fmha forward split-kv + combine kernels (#1338)

2024-06-26 17:41:15 +08:00

fmha_fwd_splitkv_tile_partitioner.hpp

[CK_TILE] fmha forward split-kv + combine kernels (#1338)

2024-06-26 17:41:15 +08:00

fmha_fwd_tile_partitioner.hpp

CK Tile FA Training kernels (#1286)

2024-06-04 13:12:45 -05:00