[AutoScheduler] Add task scheduler #6663
Merged
Conversation
This file contains 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
merrymercy
force-pushed
the
pr-task-scheduler
branch
from
570e525 to
4cccb5d
Compare
merrymercy
force-pushed
the
pr-task-scheduler
branch
4 times, most recently
from
050e351 to
cfd4b27
Compare
merrymercy
force-pushed
the
pr-task-scheduler
branch
from
cfd4b27 to
316b31e
Compare
comaniac
requested changes
Oct 12, 2020
| if cost < self.best_costs[task_idx]: | ||
| self.best_costs[task_idx] = cost | ||
|
|
||
| if len(measure_inputs) == 0: |
There was a problem hiding this comment.
Suggested change
| if len(measure_inputs) == 0: | |
| if not measure_inputs: |
There was a problem hiding this comment.
measure_inputs is not a python list. It is a tvm::runtime::Array, so we cannot use this idiom.
| random_states = &temp_random_states; | ||
| } else { | ||
| random_states->clear(); | ||
| std::pair<Array<MeasureInput>, Array<MeasureResult>> SketchPolicyNode::ContinueSearchOneRound( |
There was a problem hiding this comment.
From the functionality it doesn't look like "continue" search one round, but more like SearchOneRound. On the other hand, the current search one round it is calling is more like TuneStates.
There was a problem hiding this comment.
I still prefer the current names.
jcf94
requested changes
Oct 13, 2020
merrymercy
force-pushed
the
pr-task-scheduler
branch
from
a0cb698 to
364a7f1
Compare
merrymercy
force-pushed
the
pr-task-scheduler
branch
2 times, most recently
from
a341e4b to
c5b7842
Compare
merrymercy
force-pushed
the
pr-task-scheduler
branch
from
c5b7842 to
d43e22c
Compare
trevor-m
pushed a commit
to trevor-m/tvm
that referenced
this pull request
Oct 29, 2020
* Add task scheduler * fix lint * fix tests * fix tests * fix tests * fix test cases * fix test cases * fix tests * address comments
trevor-m
pushed a commit
to trevor-m/tvm
that referenced
this pull request
Dec 2, 2020
* Add task scheduler * fix lint * fix tests * fix tests * fix tests * fix test cases * fix test cases * fix tests * address comments
trevor-m
pushed a commit
to trevor-m/tvm
that referenced
this pull request
Dec 4, 2020
* Add task scheduler * fix lint * fix tests * fix tests * fix tests * fix test cases * fix test cases * fix tests * address comments
trevor-m
pushed a commit
to neo-ai/tvm
that referenced
this pull request
Dec 4, 2020
* Add task scheduler * fix lint * fix tests * fix tests * fix tests * fix test cases * fix test cases * fix tests * address comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
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.
python/tvm/auto_scheduler/task_scheduler.py) to allocate time resources for optimizing all subgraphs in a DNN. It prioritizes important subgraphs (e.g. performance bottleneck, graphs with a large room for improvement) according to a given objective function.ContinueSearchOneRoundto the search policy to support the task scheduler.