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

[AUTOTVM] Automated Operator Optimization #1311

Closed
merrymercy opened this issue Jun 21, 2018 · 1 comment
Closed

[AUTOTVM] Automated Operator Optimization #1311

merrymercy opened this issue Jun 21, 2018 · 1 comment

Comments

@merrymercy
Copy link
Member

merrymercy commented Jun 21, 2018

This is the tracking issue for bringing auto-tuning module to tvm. The auto-tuning module is critical to the performance and portability of tvm.
Since this is a big change, I will send several PRs to complete the integration. Here is the roadmap.

Step 1: Basic auto tuning infrastructure (#1312)

The core components to support auto-tuning in tvm:

  • Template and space definition
  • Measurement executor
  • Basic tuners: random tuner, gridsearch tuner, xgboost tuner
  • Tuning records/log
  • Tutorial on writing tunable schedule

Step 2: Rewriting TOPI with autotvm template (#1487)

  • Rewrite performance-bottleneck operators (e.g. conv2d) in tunable style
    Currently we store parameters of schedule on source file. After this step, we will release a
    log file which stores the parameters for common shapes.
  • Add automated space generation for some backends
  • Tutorial

Step3: Tuning a whole NNVM graph

  • Tune a whole nnvm graph with easy API
  • Tutorial
@tqchen tqchen changed the title [AUTOTVM] Auto-tuning module for tvm [AUTOTVM] Automated Operator Optimization Jul 25, 2018
@tqchen
Copy link
Member

tqchen commented Aug 8, 2018

Close this as most two steps are done

@tqchen tqchen closed this as completed Aug 8, 2018
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

2 participants