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

[Relay][VM] Support execution on devices #3678

Merged
merged 6 commits into from
Aug 1, 2019
Merged

Conversation

wweic
Copy link
Contributor

@wweic wweic commented Jul 31, 2019

Add copy in various places to device so we can execute on GPU.

cc @icemelon9 @zhiics @jroesch

@wweic wweic force-pushed the wweic-cuda branch 3 times, most recently from da967a3 to ce5bf4f Compare July 31, 2019 16:38
src/runtime/vm/vm.cc Outdated Show resolved Hide resolved
include/tvm/runtime/vm.h Outdated Show resolved Hide resolved
python/tvm/relay/backend/vm.py Outdated Show resolved Hide resolved
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.

src/runtime/vm/vm.cc Outdated Show resolved Hide resolved
tests/python/relay/test_backend_vm.py Outdated Show resolved Hide resolved
python/tvm/relay/backend/vm.py Show resolved Hide resolved
@icemelon
Copy link
Member

icemelon commented Aug 1, 2019

Can we just merge test_vm and test_backend_vm together?

Copy link
Member

@icemelon icemelon left a comment

Choose a reason for hiding this comment

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

lgtm

@icemelon icemelon merged commit 5357f49 into apache:master Aug 1, 2019
@icemelon
Copy link
Member

icemelon commented Aug 1, 2019

Thanks @wweic @zhiics

@wweic wweic deleted the wweic-cuda branch August 2, 2019 02:37
wweic added a commit to wweic/tvm that referenced this pull request Aug 9, 2019
* [Relay][VM] Support execution on devices

* Reduce Copy calls

* Cleanup

* Lint

* CR comments

* Merge test into test_vm.py
wweic added a commit to neo-ai/tvm that referenced this pull request Sep 6, 2019
* [Relay][VM] Support execution on devices

* Reduce Copy calls

* Cleanup

* Lint

* CR comments

* Merge test into test_vm.py
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants