kernel
danieldk HF Staff commited on
Commit
498629c
·
1 Parent(s): 3626916

Use standard capabilities for the moe/fp8/activation kernels

Browse files
Files changed (1) hide show
  1. build.toml +0 -3
build.toml CHANGED
@@ -11,7 +11,6 @@ include = ["."]
11
  pyext = ["py", "json"]
12
 
13
  [kernel.fp8]
14
- cuda-capabilities = ["7.0", "7.2", "7.5", "8.0", "8.6", "8.7", "8.9", "9.0", "10.0", "10.1", "12.0"]
15
  src = [
16
  "cuda_compat.h",
17
  "dispatch_utils.h",
@@ -26,7 +25,6 @@ depends = ["torch"]
26
 
27
 
28
  [kernel.moe]
29
- cuda-capabilities = ["7.0", "7.2", "7.5", "8.0", "8.6", "8.7", "8.9", "9.0", "10.0", "10.1", "12.0"]
30
  src = [
31
  "cuda_compat.h",
32
  "dispatch_utils.h",
@@ -55,7 +53,6 @@ include = ["."]
55
  depends = ["torch"]
56
 
57
  [kernel.activation]
58
- cuda-capabilities = ["7.0", "7.2", "7.5", "8.0", "8.6", "8.7", "8.9", "9.0", "10.0", "10.1", "12.0"]
59
  src = [
60
  "activation/activation_kernels.cu",
61
  "activation/cuda_compat.h",
 
11
  pyext = ["py", "json"]
12
 
13
  [kernel.fp8]
 
14
  src = [
15
  "cuda_compat.h",
16
  "dispatch_utils.h",
 
25
 
26
 
27
  [kernel.moe]
 
28
  src = [
29
  "cuda_compat.h",
30
  "dispatch_utils.h",
 
53
  depends = ["torch"]
54
 
55
  [kernel.activation]
 
56
  src = [
57
  "activation/activation_kernels.cu",
58
  "activation/cuda_compat.h",