Fix capabilities
Browse files- build.toml +1 -1
build.toml
CHANGED
|
@@ -38,7 +38,7 @@ src = [
|
|
| 38 |
depends = ["torch"]
|
| 39 |
|
| 40 |
[kernel.moe-marlin]
|
| 41 |
-
cuda-capabilities = ["
|
| 42 |
src = [
|
| 43 |
"core/exception.hpp",
|
| 44 |
"core/scalar_type.hpp",
|
|
|
|
| 38 |
depends = ["torch"]
|
| 39 |
|
| 40 |
[kernel.moe-marlin]
|
| 41 |
+
cuda-capabilities = ["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",
|