-
Notifications
You must be signed in to change notification settings - Fork 3.5k
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
Conversation
There was a problem hiding this 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
python/tvm/micro/project.py
Outdated
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], |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
mlf: typing.Union[pathlib.Path, str], | |
mlf_path: typing.Union[pathlib.Path, str], |
python/tvm/micro/project.py
Outdated
project_dir : pathlib.Path or str | ||
Path to a directory where the project will be created. | ||
|
||
mlf : pathlib.Path or str |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
mlf : pathlib.Path or str | |
mlf_path : pathlib.Path or str |
python/tvm/micro/project.py
Outdated
mlf: typing.Union[pathlib.Path, str], | ||
options: dict, | ||
): | ||
"""Generate a project from a platform template and an existing MLF. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
"""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. |
2f06922
to
7bdd26c
Compare
@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>
7bdd26c
to
7f6da2c
Compare
@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 thanks! |
* 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) ...
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>
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>
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