Use standard capabilities for the moe/fp8/activation kernels
Browse files- 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",
|