Precompute mappings between M <-> D sparse representations.

PiperOrigin-RevId: 670898353 Change-Id: I4b15477b1b6eb4a64cf7d4a0bb5af8fcb95d4e45
2024-09-04 02:56:05 -07:00
parent 303e5e7718
commit 42eb669e4e
11 changed files with 156 additions and 79 deletions
@@ -4866,6 +4866,20 @@ STRUCTS: Mapping[str, StructDecl] = dict([
                 ),
                 doc='column indices of non-zeros                      (nD x 1)',  # pylint: disable=line-too-long
             ),
+             StructFieldDecl(
+                 name='mapM2D',
+                 type=PointerType(
+                     inner_type=ValueType(name='int'),
+                 ),
+                 doc='index mapping from M to D                        (nD x 1)',  # pylint: disable=line-too-long
+             ),
+             StructFieldDecl(
+                 name='mapD2M',
+                 type=PointerType(
+                     inner_type=ValueType(name='int'),
+                 ),
+                 doc='index mapping from D to M                        (nM x 1)',  # pylint: disable=line-too-long
+             ),
             StructFieldDecl(
                 name='B_rownnz',
                 type=PointerType(