merrymercy commented on a change in pull request #6310:
URL: https://github.com/apache/incubator-tvm/pull/6310#discussion_r476790568
##########
File path: src/auto_scheduler/search_policy/sketch_policy.cc
##########
@@ -363,8 +376,150 @@ Array<State> SketchPolicyNode::EvolutionarySearch(const
Array<State>& init_popul
Array<State> best_states;
auto tic_begin = std::chrono::high_resolution_clock::now();
- // TODO(comaniac, merrymercy, jcf94): Since we haven't finished porting the
cost model part
- // yet, currently delete the implementation of EvolutionarySearch. To be
added later.
+ size_t population = init_population.size();
+ int num_iters =
+ static_cast<int>(GetIntParam(params,
SketchParamKey::EvolutionarySearch::num_iters));
+ double mutation_prob = static_cast<double>(
+ GetDoubleParam(params,
SketchParamKey::EvolutionarySearch::mutation_prob));
+
+ // Two ping pong buffers to avoid copy.
+ Array<State> states_buf1{init_population}, states_buf2;
+ states_buf1.reserve(population);
+ states_buf2.reserve(population);
+ Array<State>* pnow = &states_buf1;
+ Array<State>* pnext = &states_buf2;
+
+ // The set of explored states to avoid redendants.
Review comment:
```suggestion
// The set of explored states to avoid redundancy.
```
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
