[looptree] Finish test for symbolic reuse analysis

Author: Michael Gilbert

tests/looptree/test_symbolic_reuse_analysis.py

@@ -1,9 +1,7 @@
 import unittest
-from pathlib import Path
-from itertools import starmap
+from sympy import ceiling
 
-from bindings.looptree import *
-from tests.util import TEST_TMP_DIR
+from bindings.looptree import LooptreeWorkload, LooptreeWorkloadDependencyAnalyzer
 
 from tests.load_config_mixin import LoadConfigMixin
 
@@ -22,5 +20,20 @@ def test_model_with_two_level_mm(self):
         workload = LooptreeWorkload.parse_cfg(config.root['problem'])
         analyzer = LooptreeWorkloadDependencyAnalyzer(workload)
 
-        result = analyze_reuse(mapping, workload, analyzer)
-        print(result)
+        tile_shapes, result = analyze_reuse(mapping, workload, analyzer)
+
+        self.assertEqual(len(tile_shapes), 3)
+        P1_tile_shape, C1_tile_shape, M1_tile_shape = tile_shapes
+
+        REFERENCE_FILLS = {
+            ('DRAM', 0, 0): (None, 18),
+            ('DRAM', 1, 0): (None, 8),
+            ('DRAM', 2, 0): (None, 36),
+            ('GlobalBuffer', 0, 0): (None, 18.0*ceiling(4/M1_tile_shape)),
+            ('GlobalBuffer', 1, 0): (None, 8)
+        }
+
+        for key, ref_value in REFERENCE_FILLS.items():
+            self.assertEqual(result.fills[key],
+                             ref_value,
+                             f'fills for {key} do not match')

tests/test_configs/symbolic-mapping.yaml

@@ -2,23 +2,23 @@ mapping:
   type: fused
   nodes:
   - type: storage
-    target: 0
+    target: DRAM
     dspace: [Filter1, Fmap1, Fmap2]
   - type: storage
-    target: 1
+    target: GlobalBuffer
     dspace: [Filter1]
   - type: temporal
     rank: P1
   - type: storage
-    target: 2
+    target: GlobalBuffer
     dspace: [Fmap2]
   - type: temporal
     rank: C1
   - type: spatial
     rank: M1
   - type: storage
-    target: 3
+    target: GlobalBuffer
     dspace: [Fmap1]
   - type: compute
     einsum: Fc1
-    target: 4
+    target: MAC