Hugging Face's logo

kernels-community
/
flash-mla

kernel
flash-mla
deepseek
kernel-builder
Model card Files Community
flash-mla
File size: 506 Bytes 
1f83cde
 
 
 
 
 
 
 
8fcb5e5
067c8fe
8fcb5e5
068b12a
 
1f83cde
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
 
---
tags:
- kernel
- flash-mla
- deepseek
- kernel-builder
---

![Status](https://hubwebhook.dholtz.com/shield?repo=kernels-community/flash-mla)
![Status](https://hubwebhook.dholtz.com/shield?repo=kernels-community/flash-mla-arm)

## flash-mla 

This repo builds Deepseeks [FlashMLA](https://github.com/deepseek-ai/FlashMLA) kernel via the HF [kernel-builder](https://github.com/huggingface/kernel-builder)

### Dev
```bash
nix develop -L
pytest -vv tests/
```

### Build
```bash
nix build .#bundle -L
```