Po Yen Chen 92d45d1681 Fix wrong fp8 QK/KV block gemm setting

2024-04-08 12:39:17 +00:00

..

block_fmha_pipeline_problem.hpp

Add element function to fmha api

2024-03-29 18:05:36 -04:00

block_fmha_pipeline_qr_ks_vs_async_default_policy.hpp

fix build wip

2024-02-29 22:27:31 +00:00

block_fmha_pipeline_qr_ks_vs_async.hpp

Fix bug of elementwise op, our elementwise op is not inout

2024-04-04 03:17:36 +00:00

block_fmha_pipeline_qr_ks_vs_default_policy.hpp

fix build wip

2024-02-29 22:27:31 +00:00

block_fmha_pipeline_qr_ks_vs_fp8.hpp

Remove f8 pipeline, we should share the same pipeline even in f8

2024-04-08 09:56:23 +00:00

block_fmha_pipeline_qr_ks_vs.hpp

Fix bug of elementwise op, our elementwise op is not inout

2024-04-04 03:17:36 +00:00

block_fmha_pipeline_qs_ks_vs_default_policy.hpp

fix build wip

2024-02-29 22:27:31 +00:00

block_fmha_pipeline_qs_ks_vs.hpp

Fix bug of elementwise op, our elementwise op is not inout

2024-04-04 03:17:36 +00:00

block_fmha_pipeline_qx_ks_vs_custom_policy.hpp

Fix wrong fp8 QK/KV block gemm setting

2024-04-08 12:39:17 +00:00

fmha_element_functions.hpp

To prevent compiler issue, remove the elementwise function we have not used.

2024-04-08 09:44:21 +00:00

tile_fmha_shape.hpp

fix build wip

2024-02-29 22:27:31 +00:00

tile_fmha_traits.hpp

add code

2024-02-28 22:57:19 +00:00