Commit 9fa4241
committed
Add native nvidia backend for flash attention.
1 parent f3f4bf1 commit 9fa4241
File tree
4 files changed
+544
-12
lines changed- src
- infinicore/ops/flash_attention
- infiniop/ops/flash_attention
- nvidia
4 files changed
+544
-12
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
1 | 1 | | |
2 | 2 | | |
3 | 3 | | |
| 4 | + | |
4 | 5 | | |
5 | 6 | | |
6 | 7 | | |
| |||
0 commit comments