Update: restructure the zigzag stages and add api for zigzag-imc #1

JiacongSun · 2023-11-12T18:56:20Z

Update:
Now except IMC cost models, all other features are updated to align with the base ZigZag repository. In details:

RemoveNoUseMemStage, SearchNoUseMemStage are renamed to RemoveUnusedMemoryStage, SearchUnusedMemoryStage.
autogenerator.py is deleted. It is now merged into generator.py.
cost_model_in_sram_computing.py is renamed to cost_model_for_sram_imc.py.
api.py is added under zigzag/ folder. The section of Getting Started in README.md is updated and api.py is mentioned there.

Fixes:

A bug is fixed that the generated spatial mapping will exceed the layer size when D3 > 1.
A bug is fixed that sometimes there is no spatial mapping generated when a layer dim is mapped on multiple hardware dims.

…y is to be fixed.

… dim is mapped on multiple hardware dims.

JiacongSun added 8 commits November 11, 2023 21:50

update new zigzag files to zigzag-imc. api.py is created. main_onnx.p…

1445d70

…y is to be fixed.

Fix the issue that sometimes there is no sm loop yielded when a layer…

3d5ff88

… dim is mapped on multiple hardware dims.

delete .pyc files generated by pycharm

348082a

update .gitignore and add .pyc

7e57140

delete __pycache__ files

52190b9

update .gitignore and add __pycache__ folder

6febf7b

delete swp files generated by gvim

82dcf39

remove the comments in SaveStage.py

012ad37

JiacongSun merged commit 5d71984 into KULeuven-MICAS:master Nov 13, 2023
1 check failed

Provide feedback