make visualization for example architectures optional

KULeuven-MICAS · Sep 22, 2023 · cdf0de8 · cdf0de8
1 parent e06b2ac
commit cdf0de8
Show file tree

Hide file tree

Showing 6 changed files with 36 additions and 33 deletions.
diff --git a/zigzag/inputs/examples/hardware/Ascend_like.py b/zigzag/inputs/examples/hardware/Ascend_like.py
@@ -8,7 +8,7 @@
 from zigzag.classes.hardware.architecture.core import Core
 
 
-def memory_hierarchy_dut(multiplier_array):
+def memory_hierarchy_dut(multiplier_array, visualize=False):
     """Memory hierarchy variables"""
     """ size=#bit, bw=(read bw, write bw), cost=(read word energy, write work energy) """
 
@@ -243,12 +243,12 @@ def memory_hierarchy_dut(multiplier_array):
         ),
         served_dimensions="all",
     )
+    if visualize:
+        from zigzag.visualization.graph.memory_hierarchy import (
+            visualize_memory_hierarchy_graph,
+        )
 
-    from zigzag.visualization.graph.memory_hierarchy import (
-        visualize_memory_hierarchy_graph,
-    )
-
-    # visualize_memory_hierarchy_graph(memory_hierarchy_graph)
+        visualize_memory_hierarchy_graph(memory_hierarchy_graph)
     return memory_hierarchy_graph
 
 

diff --git a/zigzag/inputs/examples/hardware/Edge_TPU_like.py b/zigzag/inputs/examples/hardware/Edge_TPU_like.py
@@ -8,7 +8,7 @@
 from zigzag.classes.hardware.architecture.core import Core
 
 
-def memory_hierarchy_dut(multiplier_array):
+def memory_hierarchy_dut(multiplier_array, visualize=False):
     """Memory hierarchy variables"""
     """ size=#bit, bw=(read bw, write bw), cost=(read word energy, write work energy) """
 
@@ -152,12 +152,12 @@ def memory_hierarchy_dut(multiplier_array):
         ),
         served_dimensions="all",
     )
+    if visualize:
+        from zigzag.visualization.graph.memory_hierarchy import (
+            visualize_memory_hierarchy_graph,
+        )
 
-    from zigzag.visualization.graph.memory_hierarchy import (
-        visualize_memory_hierarchy_graph,
-    )
-
-    # visualize_memory_hierarchy_graph(memory_hierarchy_graph)
+        visualize_memory_hierarchy_graph(memory_hierarchy_graph)
     return memory_hierarchy_graph
 
 

diff --git a/zigzag/inputs/examples/hardware/Eyeriss_like.py b/zigzag/inputs/examples/hardware/Eyeriss_like.py
@@ -7,7 +7,7 @@
 from zigzag.classes.hardware.architecture.core import Core
 
 
-def memory_hierarchy_latency_test1(multiplier_array):
+def memory_hierarchy_latency_test1(multiplier_array, visualize=False):
     """Memory hierarchy variables"""
     """ size=#bit, bw=(read bw, write bw), cost=(read word energy, write work energy) """
     rf1 = MemoryInstance(
@@ -167,9 +167,12 @@ def memory_hierarchy_latency_test1(multiplier_array):
         ),
         served_dimensions="all",
     )
+    if visualize:
+        from zigzag.visualization.graph.memory_hierarchy import (
+            visualize_memory_hierarchy_graph,
+        )
 
-    # from visualization.graph.memory_hierarchy import visualize_memory_hierarchy_graph
-    # visualize_memory_hierarchy_graph(memory_hierarchy_graph)
+        visualize_memory_hierarchy_graph(memory_hierarchy_graph)
     return memory_hierarchy_graph
 
 

diff --git a/zigzag/inputs/examples/hardware/Meta_prototype.py b/zigzag/inputs/examples/hardware/Meta_prototype.py
@@ -8,7 +8,7 @@
 from zigzag.classes.hardware.architecture.core import Core
 
 
-def memory_hierarchy_dut(multiplier_array):
+def memory_hierarchy_dut(multiplier_array, visualize=False):
     """Memory hierarchy variables"""
     """ size=#bit, bw=(read bw, write bw), cost=(read word energy, write work energy) """
 
@@ -197,12 +197,12 @@ def memory_hierarchy_dut(multiplier_array):
         ),
         served_dimensions="all",
     )
+    if visualize:
+        from zigzag.visualization.graph.memory_hierarchy import (
+            visualize_memory_hierarchy_graph,
+        )
 
-    from zigzag.visualization.graph.memory_hierarchy import (
-        visualize_memory_hierarchy_graph,
-    )
-
-    # visualize_memory_hierarchy_graph(memory_hierarchy_graph)
+        visualize_memory_hierarchy_graph(memory_hierarchy_graph)
     return memory_hierarchy_graph
 
 

diff --git a/zigzag/inputs/examples/hardware/TPU_like.py b/zigzag/inputs/examples/hardware/TPU_like.py
@@ -8,7 +8,7 @@
 from zigzag.classes.hardware.architecture.core import Core
 
 
-def memory_hierarchy_dut(multiplier_array):
+def memory_hierarchy_dut(multiplier_array, visualize=False):
     """Memory hierarchy variables"""
     """ size=#bit, bw=(read bw, write bw), cost=(read word energy, write work energy) """
 
@@ -130,12 +130,12 @@ def memory_hierarchy_dut(multiplier_array):
         ),
         served_dimensions="all",
     )
+    if visualize:
+        from zigzag.visualization.graph.memory_hierarchy import (
+            visualize_memory_hierarchy_graph,
+        )
 
-    from zigzag.visualization.graph.memory_hierarchy import (
-        visualize_memory_hierarchy_graph,
-    )
-
-    visualize_memory_hierarchy_graph(memory_hierarchy_graph)
+        visualize_memory_hierarchy_graph(memory_hierarchy_graph)
     return memory_hierarchy_graph
 
 

diff --git a/zigzag/inputs/examples/hardware/Tesla_NPU_like.py b/zigzag/inputs/examples/hardware/Tesla_NPU_like.py
@@ -8,7 +8,7 @@
 from zigzag.classes.hardware.architecture.core import Core
 
 
-def memory_hierarchy_dut(multiplier_array):
+def memory_hierarchy_dut(multiplier_array, visualize=False):
     """Memory hierarchy variables"""
     """ size=#bit, bw=(read bw, write bw), cost=(read word energy, write work energy) """
 
@@ -200,12 +200,12 @@ def memory_hierarchy_dut(multiplier_array):
         ),
         served_dimensions="all",
     )
+    if visualize:
+        from zigzag.visualization.graph.memory_hierarchy import (
+            visualize_memory_hierarchy_graph,
+        )
 
-    from zigzag.visualization.graph.memory_hierarchy import (
-        visualize_memory_hierarchy_graph,
-    )
-
-    # visualize_memory_hierarchy_graph(memory_hierarchy_graph)
+        visualize_memory_hierarchy_graph(memory_hierarchy_graph)
     return memory_hierarchy_graph