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][zephyr] Add support for host-driven AoT execution on zephyr #11650

Merged
merged 4 commits into from
Jun 14, 2022

Conversation

alanmacd
Copy link
Contributor

@alanmacd alanmacd commented Jun 9, 2022

  • add support for host-driven AoT execution on zephyr;
  • add initial version of reference counting to prevent python code from inadvertently freeing tensors during garbage collection;
  • add support for numerical indices to host-drive AoT get_input();
  • add two initial tests for host-driven AoT execution on zephyr;
  • rename existing zephyr AoT exec. test;

#11121

cc @gromero @mehrdadh @areusch @mkatanbaf

- add initial version of reference counting to prevent python code from inadvertently freeing tensors during garbage collection;
- add support for numerical indices to host-drive AoT get_input();
- add two initial tests for host-driven AoT execution on zephyr;
- rename existing zephyr AoT exec. test;
@alanmacd alanmacd marked this pull request as ready for review June 9, 2022 20:10
@alanmacd alanmacd changed the title Add support for host-driven AoT execution on zephyr [microTVM][zephyr]Add support for host-driven AoT execution on zephyr Jun 9, 2022
@alanmacd alanmacd changed the title [microTVM][zephyr]Add support for host-driven AoT execution on zephyr [microTVM][zephyr] Add support for host-driven AoT execution on zephyr Jun 9, 2022
Copy link
Contributor

@areusch areusch left a comment

Choose a reason for hiding this comment

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

thanks @alanmacd this looks pretty good! just a couple minor suggestions here.

python/tvm/micro/session.py Outdated Show resolved Hide resolved
python/tvm/runtime/ndarray.py Outdated Show resolved Hide resolved
tests/micro/zephyr/test_zephyr_aot_exec.py Outdated Show resolved Hide resolved
@areusch areusch merged commit 5b3cef3 into apache:main Jun 14, 2022
@areusch
Copy link
Contributor

areusch commented Jun 14, 2022

thanks @alanmacd !

@alanmacd alanmacd deleted the zephyr-host-driven-aot branch June 14, 2022 22:46
juda pushed a commit to juda/tvm that referenced this pull request Jun 21, 2022
apache#11650)

* - add support for host-driven AoT execution on zephyr;
- add initial version of reference counting to prevent python code from inadvertently freeing tensors during garbage collection;
- add support for numerical indices to host-drive AoT get_input();
- add two initial tests for host-driven AoT execution on zephyr;
- rename existing zephyr AoT exec. test;

* address PR feedback

* increase stack size to accommodate qemu_riscv64 stack usage
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

2 participants