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][Runtime] Add set_input/output_zero_copy in python #13623

Merged
merged 16 commits into from
Dec 17, 2022

Conversation

shingjan
Copy link

As titled, this apis is in cpp but not python. Added this apis for GraphModule and some tests.

cc: @vinx13

@tvm-bot
Copy link
Collaborator

tvm-bot commented Dec 15, 2022

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

python/tvm/contrib/graph_executor.py Outdated Show resolved Hide resolved
src/runtime/graph_executor/graph_executor.cc Outdated Show resolved Hide resolved
python/tvm/contrib/graph_executor.py Outdated Show resolved Hide resolved
@shingjan shingjan requested a review from vinx13 December 15, 2022 18:54
@shingjan shingjan requested a review from vinx13 December 15, 2022 19:29
@shingjan
Copy link
Author

@tvm-bot re-run

@shingjan shingjan requested a review from echuraev December 16, 2022 22:09
@masahi masahi merged commit 564934d into apache:main Dec 17, 2022
fzi-peccia pushed a commit to fzi-peccia/tvm that referenced this pull request Mar 27, 2023
…3623)

* add set_output and test for set_output_zero_copy in python

* clean up

* clean up test

* test finished

* remove set output

* remove setoutput from header

* use zero copy for params

* fix typo

* address comments

* address comments

* add second test for set_input params

* add requires_torch

* add requires torch

* remove pytest

* add error handling for c graph executor

* better handling
mikeseven pushed a commit to mikeseven/tvm that referenced this pull request Sep 27, 2023
…3623)

* add set_output and test for set_output_zero_copy in python

* clean up

* clean up test

* test finished

* remove set output

* remove setoutput from header

* use zero copy for params

* fix typo

* address comments

* address comments

* add second test for set_input params

* add requires_torch

* add requires torch

* remove pytest

* add error handling for c graph executor

* better handling
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