kernels-community
/

megablocks

Model card Files Files and versions

ahadnagy commited on 10 days ago

Commit

2949b38

·

verified ·

1 Parent(s): 5fdd4b9

Update build.toml

Support GB10 (DGX Spark).

Files changed (1) hide show

build.toml +2 -1

build.toml CHANGED Viewed

@@ -22,7 +22,8 @@ cuda-capabilities = [
     "10.0",
     "10.1",
     "11.8",
-    "12.0"
 ]
 depends = ["torch", "cutlass_3_8"]
 src = [

     "10.0",
     "10.1",
     "11.8",
+    "12.0",
+    "12.1"
 ]
 depends = ["torch", "cutlass_3_8"]
 src = [