| __version__ = "1.0.0" | |
| from flash_mla.flash_mla_interface import ( | |
| get_mla_metadata, | |
| flash_mla_with_kvcache, | |
| flash_attn_varlen_func, | |
| flash_attn_varlen_qkvpacked_func, | |
| flash_attn_varlen_kvpacked_func, | |
| flash_mla_sparse_fwd | |
| ) | |
| __all__ = [ | |
| "get_mla_metadata", | |
| "flash_mla_with_kvcache", | |
| "flash_attn_varlen_func", | |
| "flash_attn_varlen_qkvpacked_func", | |
| "flash_attn_varlen_kvpacked_func", | |
| "flash_mla_sparse_fwd" | |
| ] | |