Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[TVMC] Add support for the MLF to 'compile' command #8086

Merged
merged 2 commits into from
May 25, 2021
Merged

[TVMC] Add support for the MLF to 'compile' command #8086

merged 2 commits into from
May 25, 2021

Commits on May 24, 2021

  1. [TVMC] Add support for the MLF to 'compile' command 

    Add support for the Model Library Format (MLF) to 'tvmc' so users can
output compilation artifacts to a MLF archive passing the new flag
'--output-format mlf'. For instance:

$ python3 -m tvm.driver.tvmc compile ./sine_model.tflite --target="c" --output sine.tar --output-format mlf

will generate a sine.tar archive that is serialized accordingly to the
MLF.

Since the MLF is currently meant to be used only on micro targets, an
error is generated if one tries to run a MLF outside a micro context.

The micro context does not exist yet but will be later introduced as
part of the [RFC] "TVMC: Add support for µTVM".

That commit also adds 3 pytest tests to test tvmc + MLF.

Finally, it also fixes some missing periods in the 'compile' command
help sections and renames export_format to output_format so there is
no confusion with flag '--dump-code', which contains "formats to export"
in its help section.

Signed-off-by: Gustavo Romero <gustavo.romero@linaro.org>
    @gromero
    gromero committed May 24, 2021
    Configuration menu
    Copy the full SHA
    cfb7156 View commit details
    Browse the repository at this point in the history

  2. Fix missing importorskip in the import_package test 

    Fix missing importorskip() in the import_package test allowing the
test in question to be skipped when 'tflite' is not installed in the
test environment, otherwise the test will fail with:

[...]
>       archive_path = exported_tvmc_package.package_path
E       AttributeError: 'str' object has no attribute 'package_path'
    @gromero
    gromero committed May 24, 2021
    Configuration menu
    Copy the full SHA
    f88a343 View commit details
    Browse the repository at this point in the history