fix merge access map.

TiledTensor · May 30, 2024 · a378eda · a378eda
1 parent fff7f6d
commit a378eda
Showing 1 changed file with 7 additions and 9 deletions.
diff --git a/thriller-core/src/dataflow/block.rs b/thriller-core/src/dataflow/block.rs
@@ -64,15 +64,13 @@ impl ThrillerBlock {
         // If they are the same, then we can merge them into a unified access map.
 
         // TODO: Implement this function.
-        self.inputs.windows(2).for_each(|window| {
-            let (first, second) = (&window[0], &window[1]);
-            assert!(
-                first.get_access() == second.get_access(),
-                "Access maps are not the same."
-            );
-        });
-
-        self.unified_access_map = Some(self.inputs[0].get_access().as_ref().unwrap().clone());
+
+        self.merge_loops();
+        if self.loop_groups.len() == 1 {
+            self.unified_access_map = Some(self.inputs[0].get_access().as_ref().unwrap().clone());
+        } else {
+            self.unified_access_map = None;
+        }
     }
 
     pub(crate) fn get_inputs(&self) -> &Vec<Rc<AttachedEdge>> {