zigzag-dse 3.1.2__tar.gz → 3.2.0__tar.gz

zigzag_dse-3.2.0/MANIFEST.in

@@ -0,0 +1,2 @@
+include zigzag/inputs/workload/*.onnx
+recursive-include zigzag/classes/cacti *

{zigzag_dse-3.1.2/zigzag_dse.egg-info → zigzag_dse-3.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: zigzag-dse
-Version: 3.1.2
+Version: 3.2.0
 Summary: ZigZag - Deep Learning Hardware Design Space Exploration
 Author-email: Arne Symons <arne.symons@kuleuven.be>, Linyan Mei <linyan.mei@kuleuven.be>
 License: MIT License
@@ -25,7 +25,7 @@ License: MIT License
         OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
         SOFTWARE.
         
-Project-URL: Homepage, https://github.com/ZigZag-Project/zigzag
+Project-URL: Homepage, https://github.com/KULeuven-MICAS/zigzag
 Keywords: zigzag,dse,design-space-exploration,machine-learning,deep-learning,mapping
 Classifier: License :: OSI Approved :: BSD License
 Classifier: Programming Language :: Python
@@ -40,12 +40,15 @@ Requires-Dist: onnx
 Requires-Dist: tqdm
 Requires-Dist: multiprocessing_on_dill
 Requires-Dist: pyyaml
-Requires-Dist: tomli
+Requires-Dist: cerberus
 Provides-Extra: dev
 Requires-Dist: bumpver; extra == "dev"
 Requires-Dist: pip-tools; extra == "dev"
 Requires-Dist: build; extra == "dev"
 Requires-Dist: twine; extra == "dev"
+Requires-Dist: pytest; extra == "dev"
+Requires-Dist: typeguard; extra == "dev"
+Requires-Dist: ruff; extra == "dev"
 
 # ZigZag [Documentation](https://kuleuven-micas.github.io/zigzag/) [Tutorial](https://www.youtube.com/watch?v=VgUuG4QaSQQ&list=PLUi74Rw4uFDIuK_6FCF9Bv7SMJlHfG4l3&index=1)
 This repository presents the novel version of our tried-and-tested HW Architecture-Mapping Design Space Exploration (DSE) Framework for Deep Learning (DL) accelerators. ZigZag bridges the gap between algorithmic DL decisions and their acceleration cost on specialized accelerators through a fast and accurate HW cost estimation. 

{zigzag_dse-3.1.2 → zigzag_dse-3.2.0}/pyproject.toml

@@ -9,9 +9,12 @@ include = ["*"]  # ["*"] by default
 exclude = ["inputs*", "outputs*", "docs*"]
 namespaces = true  # true by default
 
+[tool.setuptools.package-data]
+"zigzag" = ["py.typed"]
+
 [project]
 name = "zigzag-dse"
-version = "3.1.2"
+version = "3.2.0"
 description = "ZigZag - Deep Learning Hardware Design Space Exploration"
 readme = "README.md"
 authors = [{ name = "Arne Symons", email = "arne.symons@kuleuven.be" }, { name = "Linyan Mei", email = "linyan.mei@kuleuven.be" }]
@@ -30,22 +33,22 @@ dependencies = [
     'tqdm',
     'multiprocessing_on_dill',
     'pyyaml',
-    'tomli',
+    'cerberus',
 ]
 requires-python = ">=3.11"
 
 
 [project.optional-dependencies]
-dev = ["bumpver", "pip-tools", "build", "twine"]
+dev = ["bumpver", "pip-tools", "build", "twine", 'pytest', 'typeguard', 'ruff']
 
 [project.urls]
-Homepage = "https://github.com/ZigZag-Project/zigzag"
+Homepage = "https://github.com/KULeuven-MICAS/zigzag"
 
 [project.scripts]
 realpython = "zigzag.__main__:main"
 
 [tool.bumpver]
-current_version = "3.1.2"
+current_version = "3.2.0"
 version_pattern = "MAJOR.MINOR.PATCH"
 commit_message = "bump version {old_version} -> {new_version}"
 commit = true

zigzag_dse-3.2.0/tests/main/test_imc/test_aimc.py

@@ -0,0 +1,34 @@
+import pytest
+
+from zigzag.api import get_hardware_performance_zigzag_imc
+
+workloads = (
+    "zigzag/inputs/workload/resnet18.onnx",
+    "zigzag/inputs/workload/resnet18.yaml",
+)
+
+# Expected energy, latency (#cycles), clk time and area for each workload defined above
+ens_lats_clks_areas = {
+    "zigzag/inputs/workload/resnet18.onnx": (4695829569.611852, 5540772.0, 14.07244, 2.3426222152749223),
+    "zigzag/inputs/workload/resnet18.yaml": (4265124724.2396007, 4907619.0, 14.07244, 2.3426222152749223),
+}
+
+
+@pytest.fixture
+def mapping():
+    return "zigzag/inputs/mapping/default_imc.yaml"
+
+
+@pytest.fixture
+def accelerator():
+    return "zigzag/inputs/hardware/aimc.yaml"
+
+
+@pytest.mark.parametrize("workload", workloads)
+def test_api(workload: str, accelerator: str, mapping: str):
+    (energy, latency, tclk, area, cmes) = get_hardware_performance_zigzag_imc(workload, accelerator, mapping)
+    (expected_energy, expected_latency, expected_tclk, expected_area) = ens_lats_clks_areas[workload]
+    assert energy == pytest.approx(expected_energy)
+    assert latency == pytest.approx(expected_latency)
+    assert tclk == pytest.approx(expected_tclk)
+    assert area == pytest.approx(expected_area)

zigzag_dse-3.2.0/tests/main/test_imc/test_dimc.py

@@ -0,0 +1,34 @@
+import pytest
+
+from zigzag.api import get_hardware_performance_zigzag_imc
+
+workloads = (
+    "zigzag/inputs/workload/resnet18.onnx",
+    "zigzag/inputs/workload/resnet18.yaml",
+)
+
+# Expected energy, latency (#cycles), clk time and area for each workload defined above
+ens_lats_clks_areas = {
+    "zigzag/inputs/workload/resnet18.onnx": (4726270705.225856, 6337852.0, 3.75708, 0.8566212024),
+    "zigzag/inputs/workload/resnet18.yaml": (4268432908.954752, 5789353.0, 3.75708, 0.8566212024),
+}
+
+
+@pytest.fixture
+def mapping():
+    return "zigzag/inputs/mapping/default_imc.yaml"
+
+
+@pytest.fixture
+def accelerator():
+    return "zigzag/inputs/hardware/dimc.yaml"
+
+
+@pytest.mark.parametrize("workload", workloads)
+def test_api(workload: str, accelerator: str, mapping: str):
+    (energy, latency, tclk, area, cmes) = get_hardware_performance_zigzag_imc(workload, accelerator, mapping)
+    (expected_energy, expected_latency, expected_tclk, expected_area) = ens_lats_clks_areas[workload]
+    assert energy == pytest.approx(expected_energy)
+    assert latency == pytest.approx(expected_latency)
+    assert tclk == pytest.approx(expected_tclk)
+    assert area == pytest.approx(expected_area)

zigzag_dse-3.2.0/tests/main/test_origin/test_ascend_like.py

@@ -0,0 +1,33 @@
+import pytest
+
+from zigzag.api import get_hardware_performance_zigzag
+
+workloads = (
+    "zigzag/inputs/workload/resnet18.onnx",
+    "zigzag/inputs/workload/resnet18.yaml",
+)
+
+# Expected energy and latency for each workload defined above
+ens_lats = {
+    "zigzag/inputs/workload/resnet18.onnx": (1860963861.6800003, 3698589.0),
+    "zigzag/inputs/workload/resnet18.yaml": (2411783423.28, 4779709.0),
+}
+
+
+@pytest.fixture
+def mapping():
+    return "zigzag/inputs/mapping/ascend_like.yaml"
+
+
+@pytest.fixture
+def accelerator():
+    return "zigzag/inputs/hardware/ascend_like.yaml"
+
+
+@pytest.mark.parametrize("workload", workloads)
+def test_api(workload: str, accelerator: str, mapping: str):
+    (energy, latency, _) = get_hardware_performance_zigzag(workload, accelerator, mapping)
+    (expected_energy, expected_latency) = ens_lats[workload]
+    print(f"'{workload}': ({energy}, {latency}),")
+    assert energy == pytest.approx(expected_energy)
+    assert latency == pytest.approx(expected_latency)

{zigzag_dse-3.1.2 → zigzag_dse-3.2.0}/tests/main/test_origin/test_edge_tpu_like.py

@@ -3,33 +3,29 @@ import pytest
 from zigzag.api import get_hardware_performance_zigzag
 
 workloads = (
-    "inputs/workload/alexnet.onnx",
-    "inputs/workload/mobilenetv2.onnx",
-    "inputs/workload/resnet18.onnx",
-    "inputs/workload/resnet18.yaml",
+    "zigzag/inputs/workload/resnet18.onnx",
+    "zigzag/inputs/workload/resnet18.yaml",
 )
 
 # Expected energy and latency for each workload defined above
 ens_lats = {
-    "inputs/workload/alexnet.onnx": (5657178248.6, 8221391.0),
-    "inputs/workload/mobilenetv2.onnx": (1680773377.4499998, 3562331.0),
-    "inputs/workload/resnet18.onnx": (1902637139.1499994, 3333310.0),
-    "inputs/workload/resnet18.yaml": (2413348670.67, 4268451.0),
+    "zigzag/inputs/workload/resnet18.onnx": (1902637139.1499994, 3333310.0),
+    "zigzag/inputs/workload/resnet18.yaml": (2413348670.67, 4268451.0),
 }
 
 
 @pytest.fixture
 def mapping():
-    return "inputs/mapping/edge_tpu_like.yaml"
+    return "zigzag/inputs/mapping/edge_tpu_like.yaml"
 
 
 @pytest.fixture
 def accelerator():
-    return "inputs/hardware/edge_tpu_like.yaml"
+    return "zigzag/inputs/hardware/edge_tpu_like.yaml"
 
 
 @pytest.mark.parametrize("workload", workloads)
-def test_api(workload, accelerator, mapping):
+def test_api(workload: str, accelerator: str, mapping: str):
     (energy, latency, cmes) = get_hardware_performance_zigzag(workload, accelerator, mapping)
     (expected_energy, expected_latency) = ens_lats[workload]
     print(f"'{workload}': ({energy}, {latency}),")

zigzag_dse-3.1.2/tests/main/test_origin/test_ascend_like.py → zigzag_dse-3.2.0/tests/main/test_origin/test_meta_prototype_like.py

@@ -3,33 +3,29 @@ import pytest
 from zigzag.api import get_hardware_performance_zigzag
 
 workloads = (
-    "inputs/workload/alexnet.onnx",
-    "inputs/workload/mobilenetv2.onnx",
-    "inputs/workload/resnet18.onnx",
-    "inputs/workload/resnet18.yaml",
+    "zigzag/inputs/workload/resnet18.onnx",
+    "zigzag/inputs/workload/resnet18.yaml",
 )
 
 # Expected energy and latency for each workload defined above
 ens_lats = {
-    "inputs/workload/alexnet.onnx": (5738188827.200001, 8696068.0),
-    "inputs/workload/mobilenetv2.onnx": (1913154775.600001, 7359658.0),
-    "inputs/workload/resnet18.onnx": (1860963861.6800003, 3698589.0),
-    "inputs/workload/resnet18.yaml": (2411783423.28, 4779709.0),
+    "zigzag/inputs/workload/resnet18.onnx": (1869036158.08, 3366695.0),
+    "zigzag/inputs/workload/resnet18.yaml": (2418511845.2400002, 4130645.0),
 }
 
 
 @pytest.fixture
 def mapping():
-    return "inputs/mapping/ascend_like.yaml"
+    return "zigzag/inputs/mapping/meta_prototype_like.yaml"
 
 
 @pytest.fixture
 def accelerator():
-    return "inputs/hardware/ascend_like.yaml"
+    return "zigzag/inputs/hardware/meta_prototype.yaml"
 
 
 @pytest.mark.parametrize("workload", workloads)
-def test_api(workload, accelerator, mapping):
+def test_api(workload: str, accelerator: str, mapping: str):
     (energy, latency, cmes) = get_hardware_performance_zigzag(workload, accelerator, mapping)
     (expected_energy, expected_latency) = ens_lats[workload]
     print(f"'{workload}': ({energy}, {latency}),")

{zigzag_dse-3.1.2 → zigzag_dse-3.2.0}/tests/main/test_origin/test_tesla_npu_like.py

@@ -3,33 +3,29 @@ import pytest
 from zigzag.api import get_hardware_performance_zigzag
 
 workloads = (
-    "inputs/workload/alexnet.onnx",
-    "inputs/workload/mobilenetv2.onnx",
-    "inputs/workload/resnet18.onnx",
-    "inputs/workload/resnet18.yaml",
+    "zigzag/inputs/workload/resnet18.onnx",
+    "zigzag/inputs/workload/resnet18.yaml",
 )
 
 # Expected energy and latency for each workload defined above
 ens_lats = {
-    "inputs/workload/alexnet.onnx": (6131856327.976001, 8442657.0),
-    "inputs/workload/mobilenetv2.onnx": (1683001389.998, 2824122.0),
-    "inputs/workload/resnet18.onnx": (1863716799.84, 3380242.0),
-    "inputs/workload/resnet18.yaml": (2374655424.176, 4066942.0),
+    "zigzag/inputs/workload/resnet18.onnx": (1863716799.84, 3380242.0),
+    "zigzag/inputs/workload/resnet18.yaml": (2374655424.176, 4066942.0),
 }
 
 
 @pytest.fixture
 def mapping():
-    return "inputs/mapping/tesla_npu_like.yaml"
+    return "zigzag/inputs/mapping/tesla_npu_like.yaml"
 
 
 @pytest.fixture
 def accelerator():
-    return "inputs/hardware/tesla_npu_like.yaml"
+    return "zigzag/inputs/hardware/tesla_npu_like.yaml"
 
 
 @pytest.mark.parametrize("workload", workloads)
-def test_api(workload, accelerator, mapping):
+def test_api(workload: str, accelerator: str, mapping: str):
     (energy, latency, cmes) = get_hardware_performance_zigzag(workload, accelerator, mapping)
     (expected_energy, expected_latency) = ens_lats[workload]
     print(f"'{workload}': ({energy}, {latency}),")

{zigzag_dse-3.1.2 → zigzag_dse-3.2.0}/tests/main/test_origin/test_tpu_like.py

@@ -3,33 +3,29 @@ import pytest
 from zigzag.api import get_hardware_performance_zigzag
 
 workloads = (
-    "inputs/workload/alexnet.onnx",
-    "inputs/workload/mobilenetv2.onnx",
-    "inputs/workload/resnet18.onnx",
-    "inputs/workload/resnet18.yaml",
+    "zigzag/inputs/workload/resnet18.onnx",
+    "zigzag/inputs/workload/resnet18.yaml",
 )
 
 # Expected energy and latency for each workload defined above
 ens_lats = {
-    "inputs/workload/alexnet.onnx": (5567501203.632, 9061821.0),
-    "inputs/workload/mobilenetv2.onnx": (1904482765.907, 23101112.0),
-    "inputs/workload/resnet18.onnx": (1795904402.5120003, 4158539.0),
-    "inputs/workload/resnet18.yaml": (2296490149.296, 4906975.0),
+    "zigzag/inputs/workload/resnet18.onnx": (1795904402.5120003, 4158539.0),
+    "zigzag/inputs/workload/resnet18.yaml": (2296490149.296, 4906975.0),
 }
 
 
 @pytest.fixture
 def mapping():
-    return "inputs/mapping/tpu_like.yaml"
+    return "zigzag/inputs/mapping/tpu_like.yaml"
 
 
 @pytest.fixture
 def accelerator():
-    return "inputs/hardware/tpu_like.yaml"
+    return "zigzag/inputs/hardware/tpu_like.yaml"
 
 
 @pytest.mark.parametrize("workload", workloads)
-def test_api(workload, accelerator, mapping):
+def test_api(workload: str, accelerator: str, mapping: str):
     (energy, latency, _) = get_hardware_performance_zigzag(workload, accelerator, mapping)
     (expected_energy, expected_latency) = ens_lats[workload]
     print(f"{workload}: ({energy}, {latency}),")

{zigzag_dse-3.1.2 → zigzag_dse-3.2.0}/tests/main/test_with_mix_spatial_mapping/test_ascend_like.py

@@ -7,18 +7,14 @@ from zigzag.api import get_hardware_performance_zigzag_with_mix_spatial_mapping
 # Test case for when spatial_mapping is provided, while spatial_mapping_hint is not provided.
 
 workloads = (
-    "zigzag/inputs/examples/workload/alexnet.onnx",
-    "zigzag/inputs/examples/workload/mobilenetv2.onnx",
-    "zigzag/inputs/examples/workload/resnet18.onnx",
-    "zigzag.inputs.examples.workload.resnet18",
+    "zigzag/inputs/workload/resnet18.onnx",
+    "zigzag/inputs/workload/resnet18.yaml",
 )
 
 # Expected energy and latency for each workload defined above
 ens_lats = {
-    "zigzag/inputs/examples/workload/alexnet.onnx": (5667407342.66, 8528846),
-    "zigzag/inputs/examples/workload/mobilenetv2.onnx": (921552096.0700004, 3828967),
-    "zigzag/inputs/examples/workload/resnet18.onnx": (1679218425.5100002, 3713386),
-    "zigzag.inputs.examples.workload.resnet18": (2290766279.31, 4442443),
+    "zigzag/inputs/workload/resnet18.onnx": (1679218425.5100002, 3713386),
+    "zigzag/inputs/workload/resnet18.yaml": (2290766279.31, 4442443),
 }
 
 
@@ -56,7 +52,7 @@ def accelerator():
 
 
 @pytest.mark.parametrize("workload", workloads)
-def test_api(workload, accelerator, mapping):
+def test_api(workload: str, accelerator: str, mapping: str):
     (energy, latency, cmes) = get_hardware_performance_zigzag_with_mix_spatial_mapping(workload, accelerator, mapping)
     (expected_energy, expected_latency) = ens_lats[workload]
     assert energy == pytest.approx(expected_energy)

{zigzag_dse-3.1.2 → zigzag_dse-3.2.0}/tests/main/test_with_mix_spatial_mapping/test_edge_tpu_like.py

@@ -7,18 +7,14 @@ from zigzag.api import get_hardware_performance_zigzag_with_mix_spatial_mapping
 # Test case for when both spatial_mapping and spatial_mapping_hint are provided.
 
 workloads = (
-    "zigzag/inputs/examples/workload/alexnet.onnx",
-    "zigzag/inputs/examples/workload/mobilenetv2.onnx",
-    "zigzag/inputs/examples/workload/resnet18.onnx",
-    "zigzag.inputs.examples.workload.resnet18",
+    "zigzag/inputs/workload/resnet18.onnx",
+    "zigzag/inputs/workload/resnet18.yaml",
 )
 
 # Expected energy and latency for each workload defined above
 ens_lats = {
-    "zigzag/inputs/examples/workload/alexnet.onnx": (5582059481.445, 8343378),
-    "zigzag/inputs/examples/workload/mobilenetv2.onnx": (819971935.77, 2430583),
-    "zigzag/inputs/examples/workload/resnet18.onnx": (1763135800.67, 5001291),
-    "zigzag.inputs.examples.workload.resnet18": (2090252961.0700002, 5858437),
+    "zigzag/inputs/workload/resnet18.onnx": (1763135800.67, 5001291),
+    "zigzag/inputs/workload/resnet18.yaml": (2090252961.0700002, 5858437),
 }
 
 
@@ -72,7 +68,7 @@ def accelerator():
 
 
 @pytest.mark.parametrize("workload", workloads)
-def test_api(workload, accelerator, mapping):
+def test_api(workload: str, accelerator: str, mapping: str):
     (energy, latency, cmes) = get_hardware_performance_zigzag_with_mix_spatial_mapping(workload, accelerator, mapping)
     (expected_energy, expected_latency) = ens_lats[workload]
     assert energy == pytest.approx(expected_energy)

{zigzag_dse-3.1.2 → zigzag_dse-3.2.0}/tests/main/test_with_mix_spatial_mapping/test_meta_prototype_like.py

@@ -7,18 +7,14 @@ from zigzag.api import get_hardware_performance_zigzag_with_mix_spatial_mapping
 # Test case for when an incomplete spatial_mapping is provided and spatial_mapping_hint is also provided.
 
 workloads = (
-    "zigzag/inputs/examples/workload/alexnet.onnx",
-    "zigzag/inputs/examples/workload/mobilenetv2.onnx",
-    "zigzag/inputs/examples/workload/resnet18.onnx",
-    "zigzag.inputs.examples.workload.resnet18",
+    "zigzag/inputs/workload/resnet18.onnx",
+    "zigzag/inputs/workload/resnet18.yaml",
 )
 
 # Expected energy and latency for each workload defined above
 ens_lats = {
-    "zigzag/inputs/examples/workload/alexnet.onnx": (5679695605, 8299150),
-    "zigzag/inputs/examples/workload/mobilenetv2.onnx": (901092009, 2610609),
-    "zigzag/inputs/examples/workload/resnet18.onnx": (1730672410, 3262009),
-    "zigzag.inputs.examples.workload.resnet18": (2265438430, 4017227),
+    "zigzag/inputs/workload/resnet18.onnx": (1730672410, 3262009),
+    "zigzag/inputs/workload/resnet18.yaml": (2265438430, 4017227),
 }
 
 
@@ -56,7 +52,7 @@ def accelerator():
 
 
 @pytest.mark.parametrize("workload", workloads)
-def test_api(workload, accelerator, mapping):
+def test_api(workload: str, accelerator: str, mapping: str):
     (energy, latency, cmes) = get_hardware_performance_zigzag_with_mix_spatial_mapping(workload, accelerator, mapping)
     (expected_energy, expected_latency) = ens_lats[workload]
     assert energy == pytest.approx(expected_energy)

{zigzag_dse-3.1.2 → zigzag_dse-3.2.0}/tests/main/test_with_mix_spatial_mapping/test_tesla_npu_like.py

@@ -7,18 +7,14 @@ from zigzag.api import get_hardware_performance_zigzag_with_mix_spatial_mapping
 # Test case for when only spatial_mapping_hint is provided.
 
 workloads = (
-    "zigzag/inputs/examples/workload/alexnet.onnx",
-    "zigzag/inputs/examples/workload/mobilenetv2.onnx",
-    "zigzag/inputs/examples/workload/resnet18.onnx",
-    "zigzag.inputs.examples.workload.resnet18",
+    "zigzag/inputs/workload/resnet18.onnx",
+    "zigzag/inputs/workload/resnet18.yaml",
 )
 
 # Expected energy and latency for each workload defined above
 ens_lats = {
-    "zigzag/inputs/examples/workload/alexnet.onnx": (6044768678, 8370470),
-    "zigzag/inputs/examples/workload/mobilenetv2.onnx": (930702060, 1965457),
-    "zigzag/inputs/examples/workload/resnet18.onnx": (1724869681, 3257898),
-    "zigzag.inputs.examples.workload.resnet18": (2220861655, 3934616),
+    "zigzag/inputs/workload/resnet18.onnx": (1724869681, 3257898),
+    "zigzag/inputs/workload/resnet18.yaml": (2220861655, 3934616),
 }
 
 
@@ -52,7 +48,7 @@ def accelerator():
 
 
 @pytest.mark.parametrize("workload", workloads)
-def test_api(workload, accelerator, mapping):
+def test_api(workload: str, accelerator: str, mapping: str):
     (energy, latency, cmes) = get_hardware_performance_zigzag_with_mix_spatial_mapping(workload, accelerator, mapping)
     (expected_energy, expected_latency) = ens_lats[workload]
     assert energy == pytest.approx(expected_energy)

{zigzag_dse-3.1.2 → zigzag_dse-3.2.0}/tests/main/test_with_mix_spatial_mapping/test_tpu_like.py

@@ -7,18 +7,14 @@ from zigzag.api import get_hardware_performance_zigzag_with_mix_spatial_mapping
 # Test case for when more non-existent dimensions are provided in spatial_mapping_hint.
 
 workloads = (
-    "zigzag/inputs/examples/workload/alexnet.onnx",
-    "zigzag/inputs/examples/workload/mobilenetv2.onnx",
-    "zigzag/inputs/examples/workload/resnet18.onnx",
-    "zigzag.inputs.examples.workload.resnet18",
+    "zigzag/inputs/workload/resnet18.onnx",
+    "zigzag/inputs/workload/resnet18.yaml",
 )
 
 # Expected energy and latency for each workload defined above
 ens_lats = {
-    "zigzag/inputs/examples/workload/alexnet.onnx": (5468347515.336, 8210374),
-    "zigzag/inputs/examples/workload/mobilenetv2.onnx": (730691567.0230001, 3358406),
-    "zigzag/inputs/examples/workload/resnet18.onnx": (1648700710.272, 2946593),
-    "zigzag.inputs.examples.workload.resnet18": (1972279074.768, 3455539),
+    "zigzag/inputs/workload/resnet18.onnx": (1648700710.272, 2946593),
+    "zigzag/inputs/workload/resnet18.yaml": (1972279074.768, 3455539),
 }
 
 
@@ -60,7 +56,7 @@ def accelerator():
 
 
 @pytest.mark.parametrize("workload", workloads)
-def test_api(workload, accelerator, mapping):
+def test_api(workload: str, accelerator: str, mapping: str):
     (energy, latency, cmes) = get_hardware_performance_zigzag_with_mix_spatial_mapping(workload, accelerator, mapping)
     (expected_energy, expected_latency) = ens_lats[workload]
     assert energy == pytest.approx(expected_energy)

{zigzag_dse-3.1.2 → zigzag_dse-3.2.0}/tests/main/test_without_unused_memory/test_ascend_like.py

@@ -5,18 +5,14 @@ sys.path.append("../zigzag")
 from zigzag.api import get_hardware_performance_zigzag_without_unused_memory
 
 workloads = (
-    "zigzag/inputs/examples/workload/alexnet.onnx",
-    "zigzag/inputs/examples/workload/mobilenetv2.onnx",
-    "zigzag/inputs/examples/workload/resnet18.onnx",
-    "zigzag.inputs.examples.workload.resnet18",
+    "zigzag/inputs/workload/resnet18.onnx",
+    "zigzag/inputs/workload/resnet18.yaml",
 )
 
 # Expected energy and latency for each workload defined above
 ens_lats = {
-    "zigzag/inputs/examples/workload/alexnet.onnx": (5649555894.9, 8637780),
-    "zigzag/inputs/examples/workload/mobilenetv2.onnx": (1881386179.71, 6486685),
-    "zigzag/inputs/examples/workload/resnet18.onnx": (1709089377.83, 3583047),
-    "zigzag.inputs.examples.workload.resnet18": (2243493483.15, 4657130),
+    "zigzag/inputs/workload/resnet18.onnx": (1709089377.83, 3583047),
+    "zigzag/inputs/workload/resnet18.yaml": (2243493483.15, 4657130),
 }
 
 
@@ -31,7 +27,7 @@ def accelerator():
 
 
 @pytest.mark.parametrize("workload", workloads)
-def test_api(workload, accelerator, mapping):
+def test_api(workload: str, accelerator: str, mapping: str):
     (energy, latency, cmes) = get_hardware_performance_zigzag_without_unused_memory(workload, accelerator, mapping)
     (expected_energy, expected_latency) = ens_lats[workload]
     assert energy == pytest.approx(expected_energy)

{zigzag_dse-3.1.2 → zigzag_dse-3.2.0}/tests/main/test_without_unused_memory/test_edge_tpu_like.py

@@ -5,18 +5,14 @@ sys.path.append("../zigzag")
 from zigzag.api import get_hardware_performance_zigzag_without_unused_memory
 
 workloads = (
-    "zigzag/inputs/examples/workload/alexnet.onnx",
-    "zigzag/inputs/examples/workload/mobilenetv2.onnx",
-    "zigzag/inputs/examples/workload/resnet18.onnx",
-    "zigzag.inputs.examples.workload.resnet18",
+    "zigzag/inputs/workload/resnet18.onnx",
+    "zigzag/inputs/workload/resnet18.yaml",
 )
 
 # Expected energy and latency for each workload defined above
 ens_lats = {
-    "zigzag/inputs/examples/workload/alexnet.onnx": (5568602396.684999, 8134431),
-    "zigzag/inputs/examples/workload/mobilenetv2.onnx": (751128562.4699999, 2427487),
-    "zigzag/inputs/examples/workload/resnet18.onnx": (1784539639.4349997, 3176546),
-    "zigzag.inputs.examples.workload.resnet18": (2115122870.395, 3884789),
+    "zigzag/inputs/workload/resnet18.onnx": (1784539639.4349997, 3176546),
+    "zigzag/inputs/workload/resnet18.yaml": (2115122870.395, 3884789),
 }
 
 
@@ -31,7 +27,7 @@ def accelerator():
 
 
 @pytest.mark.parametrize("workload", workloads)
-def test_api(workload, accelerator, mapping):
+def test_api(workload: str, accelerator: str, mapping: str):
     (energy, latency, cmes) = get_hardware_performance_zigzag_without_unused_memory(workload, accelerator, mapping)
     (expected_energy, expected_latency) = ens_lats[workload]
     assert energy == pytest.approx(expected_energy)

{zigzag_dse-3.1.2 → zigzag_dse-3.2.0}/tests/main/test_without_unused_memory/test_meta_prototype_like.py

@@ -5,18 +5,14 @@ sys.path.append("../zigzag")
 from zigzag.api import get_hardware_performance_zigzag_without_unused_memory
 
 workloads = (
-    "zigzag/inputs/examples/workload/alexnet.onnx",
-    "zigzag/inputs/examples/workload/mobilenetv2.onnx",
-    "zigzag/inputs/examples/workload/resnet18.onnx",
-    "zigzag.inputs.examples.workload.resnet18",
+    "zigzag/inputs/workload/resnet18.onnx",
+    "zigzag/inputs/workload/resnet18.yaml",
 )
 
 # Expected energy and latency for each workload defined above
 ens_lats = {
-    "zigzag/inputs/examples/workload/alexnet.onnx": (5679695605.4400015, 8299150),
-    "zigzag/inputs/examples/workload/mobilenetv2.onnx": (901092009.6000001, 2610609),
-    "zigzag/inputs/examples/workload/resnet18.onnx": (1730672410.3200004, 3262009),
-    "zigzag.inputs.examples.workload.resnet18": (2265438430.2299995, 4017227),
+    "zigzag/inputs/workload/resnet18.onnx": (1730672410.3200004, 3262009),
+    "zigzag/inputs/workload/resnet18.yaml": (2265438430.2299995, 4017227),
 }
 
 
@@ -31,7 +27,7 @@ def accelerator():
 
 
 @pytest.mark.parametrize("workload", workloads)
-def test_api(workload, accelerator, mapping):
+def test_api(workload: str, accelerator: str, mapping: str):
     (energy, latency, cmes) = get_hardware_performance_zigzag_without_unused_memory(workload, accelerator, mapping)
     (expected_energy, expected_latency) = ens_lats[workload]
     assert energy == pytest.approx(expected_energy)