tiagomonteiro0715 commited on
Commit
4210764
·
1 Parent(s): 6f4b39f

initial softmax attention kernel

Browse files
Files changed (1) hide show
  1. build.toml +6 -9
build.toml CHANGED
@@ -1,14 +1,11 @@
 
 
 
 
1
  [general]
2
  name = "my_softmax_function"
3
- backends = [
4
- "cpu",
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 = []