commit e7bd7ec68df3ecd1770bbad52c8e3bd767bdab2a
author Peiming Liu <peiming@google.com> Fri Jun 28 16:04:31 2024 -0700
committer GitHub <noreply@github.com> Fri Jun 28 16:04:31 2024 -0700
tree c42eb6e438d3521d02cd3da148825bd4ff075f51
parent e56115af4dd2d94cf42e8c90b8804df62129b670

[mpact][sparse] run pre-sparsification-rewrite before elementwise fus… (#57)

* [mpact][sparse] run pre-sparsification-rewrite before elementwise fusion.

* address comments.

python/mpact/mpactbackend.py

diff --git a/python/mpact/mpactbackend.py b/python/mpact/mpactbackend.py
index a98bc60..425413a 100644
--- a/python/mpact/mpactbackend.py
+++ b/python/mpact/mpactbackend.py
@@ -212,6 +212,9 @@
     + ",".join(
         [
             "func.func(linalg-generalize-named-ops)",
+            # Run pre-sparsification pass to fuse convert/cast op into
+            # producer as they might hinder kernel fusions.
+            "pre-sparsification-rewrite",
             "func.func(linalg-fuse-elementwise-ops)",
             "convert-shape-to-std",
             # Propagate sparse encodings before sparsifier mini-pipeline.

test/python/sqsum.py

diff --git a/test/python/sqsum.py b/test/python/sqsum.py
index 2d21204..6f3294c 100644
--- a/test/python/sqsum.py
+++ b/test/python/sqsum.py
@@ -28,8 +28,10 @@
 
 # Run it with MPACT.
 print("mpact")
-# TODO: make this work, expose `sparse-emit-strategy=sparse-iterator` to
-# mini-pipeline.
+# Try sparsification with sparse iterator
+# TODO: will work after explicit value is specified in the encoding.
 # res = mpact_jit(net, adj_mat, use_sp_it=True)
+# print(res)
+# Try sparsification directly with scf.for/while
 res = mpact_jit(net, adj_mat)
 print(res)