Skip to content

Adding some export features

Axel Farrugia requested to merge axelfarr/aidge_core:my_export into dev

Context

This merge request only concerns Exports.

It aims to add back some control from the export itself as it adds the possibility to specify :

  • The memory section in which the feature maps are stored;
  • The main file template;
  • Whether the network contains labels for the results verification within the main file.

Modified files

Memory Section

  • forward.jinja : Add the section as attribute while creating the mem buffer.
  • scheduler_exort.py : Add the mem_section as attribute to be read by the template.
  • export_registry.py : Add the mem_section as class variable to be changed by the export.

Main file generation

  • export_registry.py : Add the template file path as class variable to be specified by the export.
  • scheduler_export.py : Add the template file path as argument to generate the main file. Note that the aidge template file will still be used by default.

Labels

  • scheduler_export.py : Add an optional argument labels to the scheduler_export() function which is then passed as attribute while generating the main file.
Edited by Axel Farrugia

Merge request reports

Loading