kernel
danieldk HF Staff commited on
Commit
dd2f0f9
·
1 Parent(s): 40d139f

Sync capabilities with upstream

Browse files
Files changed (1) hide show
  1. build.toml +2 -4
build.toml CHANGED
@@ -6,12 +6,10 @@ src = ["torch-ext/torch_binding.cpp", "torch-ext/torch_binding.h"]
6
 
7
  [kernel.flash_attn]
8
  cuda-capabilities = [
9
- # "7.0", "7.2", "7.5",
10
  "8.0",
11
- "8.6",
12
- "8.7",
13
- "8.9",
14
  "9.0",
 
 
15
  ]
16
  src = [
17
  "flash_attn/flash_api.cpp",
 
6
 
7
  [kernel.flash_attn]
8
  cuda-capabilities = [
 
9
  "8.0",
 
 
 
10
  "9.0",
11
+ "10.0",
12
+ "12.0",
13
  ]
14
  src = [
15
  "flash_attn/flash_api.cpp",