rocking a05c4ff96c [CK TILE] Use config name instead of data type in FmhaFwdTypeConfig<config> (#1731) ...

* Add data type config, Prepare to add mix precision in the future

* Fix compile error

[ROCm/composable_kernel commit: 94ae7113bd]

2024-12-10 11:36:18 +08:00

..

__init__.py

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

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

fmha_bwd.py

[CK TILE] Use config name instead of data type in FmhaFwdTypeConfig<config> (#1731)

2024-12-10 11:36:18 +08:00

fmha_fwd_appendkv.py

[CK TILE] Use config name instead of data type in FmhaFwdTypeConfig<config> (#1731)

2024-12-10 11:36:18 +08:00

fmha_fwd_splitkv.py

[CK TILE] Use config name instead of data type in FmhaFwdTypeConfig<config> (#1731)

2024-12-10 11:36:18 +08:00

fmha_fwd.py

[CK TILE] Use config name instead of data type in FmhaFwdTypeConfig<config> (#1731)

2024-12-10 11:36:18 +08:00