Fix Incorrect Spatial Mapping Saving and Update If Condition in SpatialMappingGenerateStage

KULeuven-MICAS / zigzag

HW Architecture-Mapping Design Space Exploration Framework for Deep Learning Accelerators

https://kuleuven-micas.github.io/zigzag/

MIT License

90 stars 31 forks source link

Closed JiacongSun closed 4 months ago

JiacongSun commented 5 months ago

Issue fixed on:

The user_spatial_mapping saved in the output JSON file will be incorrect when a layer dimension is mapped onto multiple hardware dimensions. This bug only affects the mapping in the output CME and has no impact on the cost estimation result.
Updated the if condition in the SpatialMappingGeneratorStage to identify cases when weight diagonal mapping is enabled and input served dimension is set.