Instructions to use Efficient-Large-Model/Sol-Attn-Kernel-Source with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Kernels
How to use Efficient-Large-Model/Sol-Attn-Kernel-Source with Kernels:
# !pip install kernels from kernels import get_kernel kernel = get_kernel("Efficient-Large-Model/Sol-Attn-Kernel-Source") - Notebooks
- Google Colab
- Kaggle
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]
|