kernels-community
/

moe

danieldk HF Staff commited on Apr 16

Commit

0adaa5d

1 Parent(s): b2bfe4f

Sync capabilities with upstream

Files changed (1) hide show

build.toml CHANGED Viewed

@@ -11,7 +11,7 @@ include = ["."]
 pyext = ["py", "json"]
 [kernel.fp8]
-cuda-capabilities = ["7.0", "7.2", "7.5", "8.0", "8.6", "8.7", "8.9", "9.0"]
 src = [
   "cuda_compat.h",
   "dispatch_utils.h",
@@ -26,7 +26,7 @@ depends = ["torch"]
 [kernel.moe]
-cuda-capabilities = ["7.0", "7.2", "7.5", "8.0", "8.6", "8.7", "8.9", "9.0"]
 src = [
   "cuda_compat.h",
   "dispatch_utils.h",
@@ -38,7 +38,7 @@ src = [
 depends = ["torch"]
 [kernel.moe-marlin]
-cuda-capabilities = ["8.0", "8.6", "8.7", "8.9", "9.0"]
 src = [
   "core/exception.hpp",
   "core/scalar_type.hpp",
@@ -55,7 +55,7 @@ include = ["."]
 depends = ["torch"]
 [kernel.activation]
-cuda-capabilities = ["7.0", "7.2", "7.5", "8.0", "8.6", "8.7", "8.9", "9.0"]
 src = [
   "activation/activation_kernels.cu",
   "activation/cuda_compat.h",

 pyext = ["py", "json"]
 [kernel.fp8]
+cuda-capabilities = ["7.0", "7.2", "7.5", "8.0", "8.6", "8.7", "8.9", "9.0", "10.0", "10.1", "12.0"]
 src = [
   "cuda_compat.h",
   "dispatch_utils.h",
 [kernel.moe]
+cuda-capabilities = ["7.0", "7.2", "7.5", "8.0", "8.6", "8.7", "8.9", "9.0", "10.0", "10.1", "12.0"]
 src = [
   "cuda_compat.h",
   "dispatch_utils.h",
 depends = ["torch"]
 [kernel.moe-marlin]
+cuda-capabilities = ["7.0", "7.2", "7.5", "8.0", "8.6", "8.7", "8.9", "9.0", "10.0", "10.1", "12.0"]
 src = [
   "core/exception.hpp",
   "core/scalar_type.hpp",
 depends = ["torch"]
 [kernel.activation]
+cuda-capabilities = ["7.0", "7.2", "7.5", "8.0", "8.6", "8.7", "8.9", "9.0", "10.0", "10.1", "12.0"]
 src = [
   "activation/activation_kernels.cu",
   "activation/cuda_compat.h",