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

[REFACTOR][RELAY] Move invoke_tvm_op and shape_func to vm dialect #5958

Merged
merged 2 commits into from
Jul 10, 2020

Conversation

zhiics
Copy link
Member

@zhiics zhiics commented Jun 29, 2020

This PR moves invoke_tvm_op and shape_func from memory dialect to vm dialect.

@icemelon9 @jroesch @wweic

@zhiics zhiics force-pushed the vm_dialect branch 2 times, most recently from 8713f6f to f1f02e2 Compare June 29, 2020 18:15
@jroesch
Copy link
Member

jroesch commented Jun 29, 2020

Why change dialects? all of this machinery is probably needed for future work people want to do such as ahead of time compilation. cc @tqchen

@zhiics
Copy link
Member Author

zhiics commented Jun 29, 2020

@jroesch aah, I am just separating them into memory and vm namespaces. For example, AllocTensor and AllocStorage are memory dialects (relay.op.memory.alloc_tensor), but ShapeOf and ShapeFunc are vm dialects as they don't handle memory (relay.op.vm.shape_func). We may add more in each categories. Does this make sense?

@tqchen
Copy link
Member

tqchen commented Jul 6, 2020

I think the separation makes sense if we agreed on the categorization. The common one should go to the memory dialect and the vm specific ones should goes into vm. @jroesch can you followup?

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.

some minor comments

python/tvm/relay/op/vm/vm.py Show resolved Hide resolved
src/relay/op/vm/vm.cc Outdated Show resolved Hide resolved
src/relay/op/vm/vm.cc Outdated Show resolved Hide resolved
@zhiics zhiics force-pushed the vm_dialect branch 2 times, most recently from cbb6821 to 8d8f5ed Compare July 9, 2020 20:31
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
Copy link
Member

@jroesch Could you take a look again?

@jroesch jroesch merged commit ba04c6a into apache:master Jul 10, 2020
@zhiics zhiics deleted the vm_dialect branch July 10, 2020 18:19
trevor-m pushed a commit to trevor-m/tvm that referenced this pull request Jul 14, 2020
…ache#5958)

* [REFACTOR][RELAY] Move invoke_tvm_op and shape_func to vm dialect

* address comments
trevor-m pushed a commit to neo-ai/tvm that referenced this pull request Jul 14, 2020
…ache#5958)

* [REFACTOR][RELAY] Move invoke_tvm_op and shape_func to vm dialect

* address comments
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.

4 participants