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.

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

[microTVM] Add wrapper for creating project using a MLF #9090

Merged
merged 1 commit into from Sep 30, 2021

Conversation

gromero
Copy link
Contributor

@gromero gromero commented Sep 23, 2021

Currently there is already a wrapper function for creating a new project
directory based on an ExportableModule, but there isn't one for creating
a new project directory based on an existing MLF archive, which is also
handy. Hence that commit adds a new wrapper for creating a project using
an existing model compiled and kept in a MLF archive.

Signed-off-by: Gustavo Romero gustavo.romero@linaro.org

@gromero
Copy link
Contributor Author

gromero commented Sep 23, 2021

cc @areusch @mehrdadh

Copy link
Member

@Mousius Mousius left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This looks pretty straight forward @gromero - a few tweaks and then it LGTM

def generate_project_from_mlf(
template_project_dir: typing.Union[pathlib.Path, str],
project_dir: typing.Union[pathlib.Path, str],
mlf: typing.Union[pathlib.Path, str],
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Suggested change
mlf: typing.Union[pathlib.Path, str],
mlf_path: typing.Union[pathlib.Path, str],

project_dir : pathlib.Path or str
Path to a directory where the project will be created.

mlf : pathlib.Path or str
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Suggested change
mlf : pathlib.Path or str
mlf_path : pathlib.Path or str

mlf: typing.Union[pathlib.Path, str],
options: dict,
):
"""Generate a project from a platform template and an existing MLF.
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Suggested change
"""Generate a project from a platform template and an existing MLF.
"""Generate a project from a platform template and an existing Model Library Format (MLF) archive.

@gromero
Copy link
Contributor Author

gromero commented Sep 23, 2021

@Mousius Thanks for the review! All changes done in 7bdd26c

Cheers

@areusch
Copy link
Contributor

areusch commented Sep 24, 2021

@gromero thanks, please fix the CI issue

Currently there is already a wrapper function for creating a new project
directory based on an ExportableModule, but there isn't one for creating
a new project directory based on an existing MLF archive, which is also
handy. Hence that commit adds a new wrapper for creating a project using
an existing model compiled and kept in a MLF archive.

Signed-off-by: Gustavo Romero <gustavo.romero@linaro.org>
Reviewed-by: Christopher Sidebottom <chris.sidebottom@arm.com>
Reviewed-by: Andrew Reusch <areusch@octoml.ai>
@gromero
Copy link
Contributor Author

gromero commented Sep 29, 2021

@areusch @Mousius Thanks for the reviews. Sorry for the respin delay. I've fixed the offending line to satisfy the linter and also fixed the type hints for the added function (s/typing.Union/Union/) I think CI haven't caught it because the code is not being used currently -- I found that when using the code on my tvmc micro work. Hopefully 7f6da2c will be all right for the CI. Cheers.

@leandron leandron merged commit 3d64783 into apache:main Sep 30, 2021
@leandron
Copy link
Contributor

Thanks @gromero @Mousius and @areusch - this is merged now.

@gromero
Copy link
Contributor Author

gromero commented Sep 30, 2021

@leandron thanks!

AndrewZhaoLuo added a commit to AndrewZhaoLuo/tvm that referenced this pull request Sep 30, 2021
* main: (80 commits)
  Introduce centralised name transformation functions (apache#9088)
  [OpenCL] Add vectorization to cuda conv2d_nhwc schedule (apache#8636)
  [6/6] Arm(R) Ethos(TM)-U NPU codegen integration with `tvmc` (apache#8854)
  [microTVM] Add wrapper for creating project using a MLF (apache#9090)
  Fix typo (apache#9156)
  [Hotfix][Testing] Wait for RPCServer to be established (apache#9150)
  Update find cublas so it search default path if needed. (apache#9149)
  [TIR][LowerMatchBuffer] Fix lowering strides when source region has higher dimension than the buffer (apache#9145)
  Fix flaky NMS test by making sure scores are unique (apache#9140)
  [Relay] Merge analysis/context_analysis.cc and transforms/device_annotation.cc (apache#9038)
  [LLVM] Make changes needed for opaque pointers (apache#9138)
  Arm(R) Ethos(TM)-U NPU codegen integration (apache#8849)
  [CI] Split Integration tests out of first phase of pipeline (apache#9128)
  [Meta Schedule][M3b] Runner (apache#9111)
  Fix Google Mock differences between Ubuntu 18.04 and 16.04 (apache#9141)
  [TIR] add loop partition hint pragma (apache#9121)
  fix things (apache#9146)
  [Meta Schedule][M3a] SearchStrategy (apache#9132)
  [Frontend][PyTorch] support for quantized conv_transpose2d op (apache#9133)
  [UnitTest] Parametrized test_conv2d_int8_intrinsics (apache#9143)
  ...
ylc pushed a commit to ylc/tvm that referenced this pull request Jan 7, 2022
Currently there is already a wrapper function for creating a new project
directory based on an ExportableModule, but there isn't one for creating
a new project directory based on an existing MLF archive, which is also
handy. Hence that commit adds a new wrapper for creating a project using
an existing model compiled and kept in a MLF archive.

Signed-off-by: Gustavo Romero <gustavo.romero@linaro.org>
Reviewed-by: Christopher Sidebottom <chris.sidebottom@arm.com>
Reviewed-by: Andrew Reusch <areusch@octoml.ai>
ylc pushed a commit to ylc/tvm that referenced this pull request Jan 13, 2022
Currently there is already a wrapper function for creating a new project
directory based on an ExportableModule, but there isn't one for creating
a new project directory based on an existing MLF archive, which is also
handy. Hence that commit adds a new wrapper for creating a project using
an existing model compiled and kept in a MLF archive.

Signed-off-by: Gustavo Romero <gustavo.romero@linaro.org>
Reviewed-by: Christopher Sidebottom <chris.sidebottom@arm.com>
Reviewed-by: Andrew Reusch <areusch@octoml.ai>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

4 participants