Skip to content

Draft: Add profiling features for onnx cleaner

matthieu marchal requested to merge feat/profiling into dev

Context

Related to aidge_onnx#83. Enhances TensorRT export module with comparative analysis of raw and cleaned models using ONNX cleaner and other frameworks.

inference_table

Current limitations: external modules (aidge_onnx/onnxsim/onnxslim) should be installed in the running container for now.

Current changes

  • Profiling API: Added getProfileData() and getAverageExecutionTime() methods to Graph class
  • Enhanced test script: automated model performance comparison against multiple cleaners (onnx cleaner, onnxsim, onnxslim) with rich output
  • Docker integration: Added CLI parameters to test target

Next steps - Docker containerization

Currently, aidge_onnx (and dependancies) are installed directly in the running container using a mounted aidge local repository.

TODO

  • Profiling API
  • Basic compare script
  • Integration strategy decision
  • Clean docker installation
  • Unit tests
  • Documentation and examples
Edited by matthieu marchal

Merge request reports

Loading