Sync capabilities with upstream
Browse files- build.toml +4 -4
build.toml
CHANGED
|
@@ -11,7 +11,7 @@ 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"]
|
| 15 |
src = [
|
| 16 |
"cuda_compat.h",
|
| 17 |
"dispatch_utils.h",
|
|
@@ -26,7 +26,7 @@ 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"]
|
| 30 |
src = [
|
| 31 |
"cuda_compat.h",
|
| 32 |
"dispatch_utils.h",
|
|
@@ -38,7 +38,7 @@ src = [
|
|
| 38 |
depends = ["torch"]
|
| 39 |
|
| 40 |
[kernel.moe-marlin]
|
| 41 |
-
cuda-capabilities = ["8.0", "8.6", "8.7", "8.9", "9.0"]
|
| 42 |
src = [
|
| 43 |
"core/exception.hpp",
|
| 44 |
"core/scalar_type.hpp",
|
|
@@ -55,7 +55,7 @@ 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"]
|
| 59 |
src = [
|
| 60 |
"activation/activation_kernels.cu",
|
| 61 |
"activation/cuda_compat.h",
|
|
|
|
| 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 |
|
| 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",
|
|
|
|
| 38 |
depends = ["torch"]
|
| 39 |
|
| 40 |
[kernel.moe-marlin]
|
| 41 |
+
cuda-capabilities = ["7.0", "7.2", "7.5", "8.0", "8.6", "8.7", "8.9", "9.0", "10.0", "10.1", "12.0"]
|
| 42 |
src = [
|
| 43 |
"core/exception.hpp",
|
| 44 |
"core/scalar_type.hpp",
|
|
|
|
| 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",
|