rocking 108436ebf9 only output the deterministic bwd kernel for aiter (#1903) ...

* only output the deterministic kernel

* Add comment

[ROCm/composable_kernel commit: e4358c01d9]

2025-02-20 04:27:01 +08:00

..

__init__.py

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

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

fmha_bwd.py

only output the deterministic bwd kernel for aiter (#1903)

2025-02-20 04:27:01 +08:00

fmha_fwd_appendkv.py

only output the deterministic bwd kernel for aiter (#1903)

2025-02-20 04:27:01 +08:00

fmha_fwd_splitkv.py

only output the deterministic bwd kernel for aiter (#1903)

2025-02-20 04:27:01 +08:00

fmha_fwd.py

only output the deterministic bwd kernel for aiter (#1903)

2025-02-20 04:27:01 +08:00