tiagomonteiro0715 commited on
Commit ·
4210764
1
Parent(s): 6f4b39f
initial softmax attention kernel
Browse files- build.toml +6 -9
build.toml
CHANGED
|
@@ -1,14 +1,11 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
[general]
|
| 2 |
name = "my_softmax_function"
|
| 3 |
-
backends = [
|
| 4 |
-
|
| 5 |
-
"cuda",
|
| 6 |
-
"metal",
|
| 7 |
-
"rocm",
|
| 8 |
-
"xpu",
|
| 9 |
-
]
|
| 10 |
|
| 11 |
[torch]
|
| 12 |
src = []
|
| 13 |
-
|
| 14 |
-
[kernel]
|
|
|
|
| 1 |
+
# Hugging Face kernel-builder manifest.
|
| 2 |
+
# Pure-Python kernel: written in the CUTLASS Python DSL and JIT-compiled on the
|
| 3 |
+
# GPU at runtime, so there are NO compiled C++/CUDA sources to build.
|
| 4 |
+
# CUTLASS targets NVIDIA only, so the sole backend is cuda.
|
| 5 |
[general]
|
| 6 |
name = "my_softmax_function"
|
| 7 |
+
backends = ["cuda"]
|
| 8 |
+
python-depends = ["nvidia-cutlass-dsl"]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 9 |
|
| 10 |
[torch]
|
| 11 |
src = []
|
|
|
|
|
|