-
Notifications
You must be signed in to change notification settings - Fork 3.9k
[Ansor][AutoTVM v2.0] Phase 0: Ansor minimum system for auto schedule generating #5962
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
Merged
Merged
Changes from 1 commit
Commits
Show all changes
80 commits
Select commit
Hold shift + click to select a range
7ee0902
Code migration Start (#1)
jcf94 9fcbf0b
Split transform_step out & Update more UTs (#3)
jcf94 f43e82f
Add search_task, measure and serialization (#4)
jcf94 e0a5ed5
Add MetaTileRewritePolicy (#5)
jcf94 359905a
Basic Python API for State (#6)
jcf94 2032a64
Add Python API: Measure & Task (#7)
jcf94 6b21dc6
Add ansor.auto_schedule() API; First AutoSchedule working version(#8)
jcf94 e52135f
Bug fix & Add python serialization API (#10)
jcf94 1fe6638
Improve code style, python wrapper and test cases (#11)
merrymercy 43d1530
fix unit tests
merrymercy f367d15
Add RPCRunner & OpenCL/CUDA test (#12)
jcf94 2bd6471
rebase to upstream/master
merrymercy c860f2c
Add Ansor basic tutorial (#13)
jcf94 f60d1a6
migrate feature extraction (#14)
merrymercy b839c0f
Add XGBModel & RPCRunnerWarpper (#15)
jcf94 cfe58d7
Migrate workload_registry.py (#16)
merrymercy 143ea45
add task scheduler (#17)
merrymercy ed075c2
Add conv2d cuda tutorial with workload registry (#18)
jcf94 74ec7d0
add tune_test.py (the old tune_wkl.py) (#19)
merrymercy cd0a516
Code refine for tune_test.py & Add a pre load callback (#20)
jcf94 3a24e49
Add python custom sketch rule (#21)
jcf94 a155c1f
Ansor Relay Integration (without layout rewrite) (#22)
minminsun 674027f
Add tune_op_subgraph.py & Some code clean for tune_network.py (#23)
jcf94 2f241ed
add explicit_unroll_max_extent (#25)
merrymercy 18d44b8
Add Index simplification & API update (#26)
jcf94 4ea6712
Update PreLoadMeasuredStates & Some bug fix (#27)
jcf94 6126cdb
Add tensorize step for loop_state (#31)
jcf94 c7364df
State python api update (#33)
jcf94 36cd9ef
kernel layout rewrite (#28)
minminsun 145e61c
[cache flush] port cache flush to ansor (#32)
FrozenGene 2c27816
Improve relay integration (#34)
merrymercy 0794875
Fix xgb error & Simplify dispatcher (#35)
merrymercy a4c4548
Rename "MetaTileRewritePolicy" to "SketchPolicy". (#36)
merrymercy 593a2c7
rebase
merrymercy 53bd591
Migrate all node::make to noderef's construct function (#37)
jcf94 8e53d12
Some lint fix & Recover the double constructor of tvm::PrimExpr (#39)
jcf94 cd5c5ad
Add MutateComputeLocation and MutateParallel in evolutionary search (…
merrymercy 5860191
Improve loop state python API (stage_tensors -> stage_ops) (#41)
merrymercy 14a19cd
ComputeDAG bug fix & Add Custom TensorCore Matmul Example (#42)
jcf94 b012e27
Rever Commits, Start to build minimum Ansor system
jcf94 d6d6b85
Code clean for minimum Ansor system
jcf94 4042cfa
Bug fix & Delete AccessAnalyzer
jcf94 7695def
Delete attachmap & Code clean
jcf94 0c200cd
Doc update
jcf94 9c35e50
Headfile update & Python doc update
jcf94 a015051
clang-format fix
jcf94 6823802
pylint fix
jcf94 a82dbb8
Update
jcf94 ac36c46
Doc update
jcf94 a62b1e0
Update
jcf94 3eac89d
Merge branch 'upstream_master' into upstream_0_new
jcf94 526cf42
Bug fix after code merge to the new master
jcf94 426ec82
clang-format fix
jcf94 907c17c
Update
jcf94 64f8f8d
Update
jcf94 1b16dd4
Update std::vector to Array; Update verbosity setting; Some commemts
jcf94 9fa897b
std::vector->Array & std::string->String
jcf94 f40c7af
Add init_state to ComputeDAG
jcf94 0a24daf
Update
jcf94 a45fd89
Update some unordered_map to Map
jcf94 bfc6663
clang-format fix
jcf94 eb02e77
Comments addressed
jcf94 cb2442f
Lint fix
jcf94 b1ca20c
Update
jcf94 49dbec6
Merge branch 'upstream_master' into upstream_0_new
jcf94 8add768
Update
jcf94 78e5313
Update
jcf94 546abbe
Update
jcf94 d418a57
Update
jcf94 8e1d65d
Update
jcf94 3a67a72
Update
jcf94 28a7b8f
Update
jcf94 1360b1b
Update
jcf94 52afe74
Rename ansor namespace to auto_schedule
jcf94 6a61fb6
Update
jcf94 3a4e5da
Rename ThreadPool to ParallelFor
jcf94 dbe019b
Add parallel_for
jcf94 1f1b878
Remove ThreadPool
jcf94 02fede9
Update python/tvm/auto_schedule/auto_schedule.py
merrymercy eea0989
trigger CI
merrymercy File filter
Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Update
- Loading branch information
commit a82dbb80758c54d6b6c55293e095a8f337a1c791
There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Uh oh!
There was an error while loading. Please reload this page.