New feature: supporting ox/oy diagonal mapping. Fix: spatial_mapping_…

…int in cost_model.py is fetched from SpatialGeneratorStage now to avoid the mapping inconsistence.
KULeuven-MICAS · Nov 10, 2023 · b9738e1 · b9738e1
1 parent dd45365
commit b9738e1
Show file tree

Hide file tree

Showing 5 changed files with 610 additions and 112 deletions.
diff --git a/zigzag/classes/cost_model/cost_model.py b/zigzag/classes/cost_model/cost_model.py
@@ -211,12 +211,14 @@ def __init__(
         accelerator,
         layer,
         spatial_mapping,
+        spatial_mapping_int,
         temporal_mapping,
         access_same_data_considered_as_no_access=True,
     ):
         self.accelerator = accelerator
         self.layer = layer
         self.spatial_mapping = spatial_mapping
+        self.spatial_mapping_int = spatial_mapping_int  # the original spatial mapping without decimal
         self.temporal_mapping = temporal_mapping
         self.access_same_data_considered_as_no_access = (
             access_same_data_considered_as_no_access
@@ -249,7 +251,7 @@ def __init__(
         # self.spatial_mapping_dict_int = spatial_mapping_fractional_to_int(
         #     self.spatial_mapping.mapping_dict_origin
         # )
-        self.spatial_mapping_dict_int = self.spatial_mapping.mapping_dict_origin
+        self.spatial_mapping_dict_int = self.spatial_mapping_int
 
         # For constructing Mapping object,  the last parameter "self.access_same_data_considered_as_no_access" is optional
         self.mapping = Mapping(