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] Add autotvm context when compile #4062

Merged
merged 1 commit into from
Oct 5, 2019
Merged

Conversation

wweic
Copy link
Contributor

@wweic wweic commented Oct 5, 2019

This speeds up mxnet resnet18 by 2X for Relay VM in c5.18xlarge. I believe this generally applies to other models. This matches with relay.build logic(https://github.com/dmlc/tvm/blob/7d71dd8bd7b9b6412568a6860161566071588055/python/tvm/relay/build_module.py#L200-L205)

cc @icemelon9 @yongwww @kevinthesun @jroesch

@icemelon
Copy link
Member

icemelon commented Oct 5, 2019

Could you rebase the pr against the latest master?

@wweic
Copy link
Contributor Author

wweic commented Oct 5, 2019

@icemelon9 rebase done.

@zhiics zhiics merged commit 7084081 into apache:master Oct 5, 2019
@wweic wweic deleted the autotvm-vm branch October 5, 2019 17:19
anijain2305 pushed a commit to anijain2305/tvm that referenced this pull request Oct 17, 2019
wweic added a commit to neo-ai/tvm that referenced this pull request Oct 18, 2019
petrex added a commit to petrex/tvm that referenced this pull request Oct 29, 2019
* master: (21 commits)
  [Fix][VM] Fix VM invoke with set_params (apache#4079)
  [QNN] Refactor fixed point multiplication in requantize (apache#4073)
  Fix match case in Python-side expr functor (apache#4037)
  Hide symbols from dependent libraries if HIDE_PRIVATE_SYMBOLS is ON. (apache#4041)
  Add gradient for log-softmax (apache#4069)
  [DOC] Fix typos in tutorials (apache#4066)
  dicrease the complexity of CalcDep from exponential to linear (apache#4053)
  [Relay][AlterOp] Minor refactor. (apache#4064)
  [Relay][AlterOp] Improving support for broadcast layout alteration. (apache#4040)
  Add parses support for zeros_like tflite operator (apache#4042)
  [Bugfix][TF] reset graph after getting tag of savedmodel (apache#4055)
  [Relay][VM] Add more passes to VMCompiler (apache#4058)
  [Relay][VM] Add autotvm context when compile (apache#4062)
  [Bugfix] Fix target host for vm compiler (apache#4057)
  [Relay][Training] Add gradient for Crossentropy (apache#3925)
  [llvm] switch to use Align for llvm trunk (apache#4051)
  [Relay][TopHub] Add switch to disable TopHub download (apache#4015)
  [Relay][Op] Add instance norm op (apache#4004)
  [QNN][Relay] Calling Dialect passes from inside Relay Build API. (apache#3971)
  [RELAY/PASS] Fix the extent for the post_stmt in the loop partition (apache#3734)
  ...
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