added apps required by microbench and sparse ml

StanfordAHA · Oct 16, 2024 · 58eacc1 · 58eacc1
1 parent 3410528
commit 58eacc1
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/aha/util/regress.py b/aha/util/regress.py
@@ -313,6 +313,7 @@ def dispatch(args, extra_args=None):
     suitesparse_data = ["rand_large_tensor5"]
     if args.config == "fast":
         width, height = 28, 16
+        # to unroll, use 12x16
         sparse_tests = [
             # "mat_elemadd3",
             # "mat_elemmul",
@@ -322,7 +323,14 @@ def dispatch(args, extra_args=None):
             # "tensor3_ttv",
             # "tensor3_ttm", # have to use different cgra size (e.g. 20x16)
             # "tensor3_innerprod",
-            "fp_matmul_ijk",
+            "fp_matmul_ijk_crddrop",
+            # "fp_matmul_ikj",
+            # "matmul_ijk_crddrop_relu",
+            # "matmul_ijk_crddrop",
+            # "spmm_ijk_crddrop_relu",
+            # "spmm_ijk_crddrop",
+            # "mat_elemadd_leakyrelu_exp",
+            # "fp_mat_elemadd",
             # "fp_spmm_ijk",
             # "tensor3_mttkrp"
         ]