tiagomonteiro0715 commited on
Commit ·
fcfc6d3
1
Parent(s): 2b070dd
initial softmax attention kernel
Browse files- build.toml +3 -0
build.toml
CHANGED
|
@@ -10,6 +10,9 @@ backends = ["cuda"]
|
|
| 10 |
# nested [general.cuda] table, NOT the top-level python-depends (which only
|
| 11 |
# resolves "general" deps like einops).
|
| 12 |
[general.cuda]
|
|
|
|
|
|
|
|
|
|
| 13 |
python-depends = ["nvidia-cutlass-dsl"]
|
| 14 |
|
| 15 |
[torch]
|
|
|
|
| 10 |
# nested [general.cuda] table, NOT the top-level python-depends (which only
|
| 11 |
# resolves "general" deps like einops).
|
| 12 |
[general.cuda]
|
| 13 |
+
# nvidia-cutlass-dsl 4.3.0 requires CUDA >= 12.8; this drops the older-CUDA
|
| 14 |
+
# build variants where the wheel is marked broken.
|
| 15 |
+
minver = "12.8"
|
| 16 |
python-depends = ["nvidia-cutlass-dsl"]
|
| 17 |
|
| 18 |
[torch]
|