composable_kernel

mirror of https://github.com/ROCm/composable_kernel.git synced 2026-04-19 22:39:03 +00:00

Files

ltqin dd249f1cd6 Add input fp8 and output bf16 attention (#2726 )

* change host using fp16 to check

* fp8 to fp8 compare

* rewrite input parameters

* add not squant

* remove some output code

* for scale = 1

* format

* saturates only for fp8

* add fp8bf16 data type

* add fp8bf16 data type

* fix test fp8 code

* add run_fp8bf16_tests

* change fmha fwd example parameter(adding fp8bf16)

* Support fp8bf16 for Aiter

* Support aiter fp8bf16 in c++

* fix comment about fp8 in readme.md

* add fp8fp32

* add fp8fp32 test

* remove range_q etc.

* format

* fix test parameters about squant and fmha example input fp8bf16 fp8fp32 data type

* add fp8bf16 to data_type function

* change colmajor to rowmajor in test_ck_tile_fmha_fwd_fp8

* format

* reset atol for fp8

* fix bug for atol

---------

Co-authored-by: rocking <ChunYu.Lai@amd.com>
Co-authored-by: asleepzzz <hanwen.chang@amd.com>

2025-09-19 14:26:43 +08:00

benchmark_bwd.sh

[CK_TILE] Add PagedAttention kernels (#1387 )

2024-08-28 20:50:43 +08:00

benchmark_fwd_v3.sh

[CK_TILE] FMHA FAv3 scheduling fine-tuning for performance (#2833 )

2025-09-16 11:32:38 +08:00

benchmark_fwd.sh

Re-enable optimization for gfx950 fmha fwd (#2671 )

2025-08-13 14:57:43 +08:00

fmha_bwd_known_fails_gfx90a.txt

[CK_TILE] FMHA Test Ignore Known Errors (#2872 )