File size: 465 Bytes
8e9f35a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
[general]
name = "sol-attn"
version = 1
edition = 5
license = "Apache-2.0"
backends = ["cuda"]
python-depends = ["tvm-ffi"]
upstream = "https://github.com/NVlabs/Sana.git"
source = "https://huggingface.co/Efficient-Large-Model/Sol-Attn-Kernel-Source"

[general.cuda]
minver = "12.8"
python-depends = ["nvidia-cutlass-dsl"]

[general.hub]
repo-id = "Efficient-Large-Model/Sol-Attn"

[torch-noarch]
cuda-capabilities = ["8.0", "8.9", "9.0", "10.0", "12.0"]

[kernel]