kernels-community
/

moe

Model card Files Files and versions Community

danieldk HF Staff commited on 19 days ago

Commit

4727baf

·

1 Parent(s): e624017

Fix capabilities

Files changed (1) hide show

build.toml +1 -1

build.toml CHANGED Viewed

@@ -38,7 +38,7 @@ src = [
 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.moe-marlin]
+cuda-capabilities = ["8.0", "8.6", "8.7", "8.9", "9.0", "10.0", "10.1", "12.0"]
 src = [
   "core/exception.hpp",
   "core/scalar_type.hpp",