carlushuang b3f86e79dd workaround rocm-6.2 compiler issue (#1421)

2024-07-31 16:03:59 +08:00

..

block_fmha_bwd_dot_do_o_default_policy.hpp

CK Tile FA Training kernels (#1286)

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

block_fmha_bwd_dot_do_o.hpp

CK Tile FA Training kernels (#1286)

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

block_fmha_bwd_dq_dk_dv_pipeline_ks_kts_vr_default_policy.hpp

CK Tile FA Training kernels (#1286)

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

block_fmha_bwd_dq_dk_dv_pipeline_ks_kts_vr.hpp

CK Tile FA Training kernels (#1286)

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

block_fmha_bwd_dq_dk_dv_pipeline_ks_vr_default_policy.hpp

CK Tile FA Training kernels (#1286)

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

block_fmha_bwd_dq_dk_dv_pipeline_ks_vr.hpp

CK Tile FA Training kernels (#1286)

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

block_fmha_bwd_dq_dk_dv_pipeline_qs_ks_vr_dos_default_policy.hpp

CK Tile FA Training kernels (#1286)

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

block_fmha_bwd_dq_dk_dv_pipeline_qs_ks_vr_dos.hpp

CK Tile FA Training kernels (#1286)

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

block_fmha_bwd_pipeline_default_policy.hpp

Fix FA bwd alibi+causal NaN errors (#1352)

2024-06-20 09:50:53 -05:00

block_fmha_bwd_pipeline_enum.hpp

CK Tile FA Training kernels (#1286)

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

block_fmha_bwd_pipeline_problem.hpp

CK Tile FA Training kernels (#1286)

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

block_fmha_fwd_splitkv_combine_pipeline_default_policy.hpp

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

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

block_fmha_fwd_splitkv_combine_pipeline.hpp

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

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

block_fmha_fwd_splitkv_pipeline_qr_ks_vs_async_default_policy.hpp

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

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

block_fmha_fwd_splitkv_pipeline_qr_ks_vs_async.hpp

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

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

block_fmha_fwd_splitkv_pipeline_qr_ks_vs_default_policy.hpp

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

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

block_fmha_fwd_splitkv_pipeline_qr_ks_vs.hpp

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

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

block_fmha_pipeline_enum.hpp

[CK_TILE] support alibi (#1269)

2024-05-07 22:32:54 +08:00

block_fmha_pipeline_problem.hpp

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

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

block_fmha_pipeline_qr_ks_vs_async_default_policy.hpp

introducing ck_tile! (#1216)

2024-04-15 19:27:12 -05:00

block_fmha_pipeline_qr_ks_vs_async.hpp

workaround rocm-6.2 compiler issue (#1421)

2024-07-31 16:03:59 +08:00

block_fmha_pipeline_qr_ks_vs_default_policy.hpp

introducing ck_tile! (#1216)

2024-04-15 19:27:12 -05:00

block_fmha_pipeline_qr_ks_vs_fp8.hpp

CK Tile FA Training kernels (#1286)

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

block_fmha_pipeline_qr_ks_vs.hpp

Hacking ck_tile fmha Dropout facility (#1344)

2024-06-19 10:37:22 +08:00

block_fmha_pipeline_qs_ks_vs_default_policy.hpp

introducing ck_tile! (#1216)

2024-04-15 19:27:12 -05:00

block_fmha_pipeline_qs_ks_vs.hpp

CK Tile FA Training kernels (#1286)

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

block_fmha_pipeline_qx_ks_vs_custom_policy.hpp

CK Tile FA Training kernels (#1286)

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

tile_fmha_shape.hpp

CK Tile FA Training kernels (#1286)

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

tile_fmha_traits.hpp

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

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