tiagomonteiro0715 commited on
Commit
2b070dd
·
1 Parent(s): 4210764

initial softmax attention kernel

Browse files
Files changed (1) hide show
  1. build.toml +5 -0
build.toml CHANGED
@@ -5,6 +5,11 @@
5
  [general]
6
  name = "my_softmax_function"
7
  backends = ["cuda"]
 
 
 
 
 
8
  python-depends = ["nvidia-cutlass-dsl"]
9
 
10
  [torch]
 
5
  [general]
6
  name = "my_softmax_function"
7
  backends = ["cuda"]
8
+
9
+ # nvidia-cutlass-dsl is a CUDA-backend-specific python dep, so it goes in the
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]