Skip to content

[microTVM] Fix host-driven AOT memory workspaces#13807

Merged
mehrdadh merged 5 commits intoapache:mainfrom
alanmacd:fix-host-aot-mem-pools
Jan 26, 2023
Merged

[microTVM] Fix host-driven AOT memory workspaces#13807
mehrdadh merged 5 commits intoapache:mainfrom
alanmacd:fix-host-aot-mem-pools

Conversation

@alanmacd
Copy link
Copy Markdown
Contributor

@alanmacd alanmacd commented Jan 18, 2023

When using host-driven AOT with memory pools enabled, the workspace and constant memory were not properly supported. In order for them to work properly, the _run function (typically tvmgen_default_run()) needed to be called instead of tvmgen_default___tvm_main__() in order to properly setup the memory workspace pointers.

fixes #13777

@tvm-bot
Copy link
Copy Markdown
Collaborator

tvm-bot commented Jan 18, 2023

Thanks for contributing to TVM! Please refer to the contributing guidelines https://tvm.apache.org/docs/contribute/ for useful information and tips. Please request code reviews from Reviewers by @-ing them in a comment.

Generated by tvm-bot

@alanmacd alanmacd changed the title Fix host-driven AOT memory workspaces [microTVM] Fix host-driven AOT memory workspaces Jan 18, 2023
@mehrdadh mehrdadh self-assigned this Jan 19, 2023
@leandron
Copy link
Copy Markdown
Contributor

Cc @ashutosh-arm @d-smirnov

Comment thread src/runtime/crt/aot_executor/aot_executor.c
Comment thread tests/python/unittest/test_micro_model_library_format.py
@mehrdadh
Copy link
Copy Markdown
Member

Overall LGTM! I'll wait for @d-smirnov before merging it. Thanks @alanmacd!!

Comment thread src/runtime/crt/aot_executor/aot_executor.c
Comment thread src/runtime/crt/aot_executor/aot_executor.c
@mehrdadh
Copy link
Copy Markdown
Member

@alanmacd @d-smirnov thanks! this PR is merged.

@mehrdadh mehrdadh merged commit 4ddb876 into apache:main Jan 26, 2023
@alanmacd alanmacd deleted the fix-host-aot-mem-pools branch January 26, 2023 23:57
fzi-peccia pushed a commit to fzi-peccia/tvm that referenced this pull request Mar 27, 2023
When using host-driven AOT with memory pools enabled, the workspace and constant memory were not properly supported. In order for them to work properly, the _run function (typically tvmgen_default_run()) needed to be called instead of tvmgen_default___tvm_main__() in order to properly setup the memory workspace pointers.

fixes apache#13777
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.

[Bug][microTVM]Enabling USMP with VWW model has error

6 participants