Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
__init__.py 240 Bytes
Newer Older
Jiashi Li's avatar
Initial commit  
Jiashi Li committed
1
2
3
4
5 
__version__ = "1.0.0"

from flash_mla.flash_mla_interface import (
    get_mla_metadata,
    flash_mla_with_kvcache,
Shengyu Liu's avatar
Reorganize files and add sparse prefill/decoding kernels on hopper  
Shengyu Liu committed
6 
    flash_mla_sparse_fwd
Jiashi Li's avatar
Initial commit  
Jiashi Li committed
7 
)
Shengyu Liu's avatar
Multiple updates and refactorings (#150)  
Shengyu Liu committed
8
9
10
11
12
13 

__all__ = [
    "get_mla_metadata",
    "flash_mla_with_kvcache",
    "flash_mla_sparse_fwd"
]