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

Add support for using the VM across the RPC boundary. #7746

Merged
merged 10 commits into from
Mar 30, 2021

Conversation

jroesch
Copy link
Member

@jroesch jroesch commented Mar 26, 2021

Refactor some of the internals of VM in order to enable use across the RPC boundary. I need to remove the use of the lib field to instead use the imports completely still.

cc @zhiics @adelbertc @tmoreau89 @tkonolige @jwfromm @michalpiszczek

@jroesch jroesch marked this pull request as ready for review March 26, 2021 21:26
Copy link
Contributor

@tkonolige tkonolige left a comment

Choose a reason for hiding this comment

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

Just a couple of small changes.

src/runtime/library_module.cc Outdated Show resolved Hide resolved
src/runtime/library_module.cc Outdated Show resolved Hide resolved
include/tvm/runtime/vm/executable.h Show resolved Hide resolved
include/tvm/runtime/vm/executable.h Show resolved Hide resolved
include/tvm/runtime/vm/executable.h Outdated Show resolved Hide resolved
include/tvm/runtime/vm/executable.h Outdated Show resolved Hide resolved
python/tvm/runtime/module.py Outdated Show resolved Hide resolved
src/runtime/vm/executable.cc Show resolved Hide resolved
src/runtime/vm/executable.cc Outdated Show resolved Hide resolved
src/runtime/vm/executable.cc Show resolved Hide resolved
src/runtime/vm/executable.cc Outdated Show resolved Hide resolved
tests/python/relay/test_vm.py Show resolved Hide resolved
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.

just a couple more things

include/tvm/runtime/vm/executable.h Outdated Show resolved Hide resolved
python/tvm/runtime/vm.py Show resolved Hide resolved
src/runtime/library_module.cc Show resolved Hide resolved
src/relay/backend/vm/compiler.cc Show resolved Hide resolved
@jroesch
Copy link
Member Author

jroesch commented Mar 26, 2021

The last commit resolves all the outstanding comments, we will wait to be on 3.7 to use type annotations and I added doc comments and clarifications.

Copy link
Member

@zhiics zhiics left a comment

Choose a reason for hiding this comment

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

LGTM. Just a minor comment.

src/runtime/vm/executable.cc Outdated Show resolved Hide resolved
@zhiics
Copy link
Member

zhiics commented Mar 29, 2021

please fix the linting error again.

@jroesch jroesch merged commit fd18751 into apache:main Mar 30, 2021
trevor-m pushed a commit to trevor-m/tvm that referenced this pull request May 6, 2021
* Get basic verison of VM RPC working

* Test case passes

* Clean up PR

* Lint

* Format

* Address Andrew R and TK feedback

* Add comment for Andrew

* Address Zhi's comment

* Format

* Fix broken test
trevor-m pushed a commit to trevor-m/tvm that referenced this pull request May 6, 2021
* Get basic verison of VM RPC working

* Test case passes

* Clean up PR

* Lint

* Format

* Address Andrew R and TK feedback

* Add comment for Andrew

* Address Zhi's comment

* Format

* Fix broken test
trevor-m pushed a commit to trevor-m/tvm that referenced this pull request May 6, 2021
* Get basic verison of VM RPC working

* Test case passes

* Clean up PR

* Lint

* Format

* Address Andrew R and TK feedback

* Add comment for Andrew

* Address Zhi's comment

* Format

* Fix broken test
trevor-m pushed a commit to neo-ai/tvm that referenced this pull request May 11, 2021
* Get basic verison of VM RPC working

* Test case passes

* Clean up PR

* Lint

* Format

* Address Andrew R and TK feedback

* Add comment for Andrew

* Address Zhi's comment

* Format

* Fix broken test
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.

5 participants