aihwkit
aihwkit copied to clipboard
Follow-up on analog info
Description and motivation
#316 introduced an analog info functionality. Going forward, we might want to enhance it further by adding the following aspects:
- Add some more summary info like remaining digital-MACs, total number of analog tile mat-vecs
- User-definable "columns" to select only some of the columns to display
- Improve on layout, e.g. the (- - - ) first entry of the tile section is a little confusing
- Count also the peripheral digital ops (e.g noise management, bound management)
- Currently only forward pass is counted, but backward / update would be interesting to add for analog training
- Add other specifics for analog training with analog optimizers (tiki-taka / mixed precision) which need additional tiles internally and additional digital ops