-
Notifications
You must be signed in to change notification settings - Fork 3.5k
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
[ManifestAlloc] Handle TupleType inputs in CheckReshapeOnly #6776
Conversation
This arises insed ManifestAllocPass inside relay.vm.compile
@icemelon9 Can you PTAL? |
tests/python/relay/test_vm.py
Outdated
y_data = np.random.uniform(size=y_shape).astype("float32") | ||
|
||
for tgt, ctx in tvm.testing.enabled_targets(): | ||
res = veval(f, (x_data, y_data)) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
use tgt
and ctx
here?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks @altanh for pointing this. I resolved this.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
lgtm
Thanks @rohanmukh @altanh @icemelon9 |
This arises insed ManifestAllocPass inside relay.vm.compile
Thanks for contributing to TVM! Please refer to guideline https://tvm.apache.org/docs/contribute/ for useful information and tips. After the pull request is submitted, please request code reviews from Reviewers by @ them in the pull request thread.