n:m:g format implementation

spcl · May 12, 2023 · f2a5aa0 · f2a5aa0
1 parent 7457a4a
commit f2a5aa0
Show file tree

Hide file tree

Showing 8 changed files with 4,226 additions and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -25,6 +25,7 @@ dependencies = [
     "coverage",
     "pytest-cov",
     "pytest-xdist",
+    "py-cpuinfo",
 ]
 
 [project.urls]

diff --git a/src/sten/__init__.py b/src/sten/__init__.py
@@ -1,3 +1,6 @@
 from .sten import *
 
 from .patches import patch
+
+from .grouped_nm.sten_impls import GroupedNMSparsifier
+from .grouped_nm.grouped_nm_tensor import GroupedNMTensor, PerfectNMTensor
diff --git a/src/sten/grouped_nm/__init__.py b/src/sten/grouped_nm/__init__.py