Ansor search 函数调用流程

Auto Schedule 源码阅读

基于 tvm v0.13.0：https://github.com/apache/tvm/tree/v0.13.0

commit：97c5de6

https://github.com/apache/tvm/blob/97c5de63ba6baa148fdf0c545ded2cd01d7838d1

• auto_scheduler.task_scheduler.TaskScheduler.tune
  • auto_scheduler.task_scheduler.TaskScheduler._tune_task(task_idx)
    • auto_scheduler.search_policy.SearchPolicy.continue_search_one_round
      • _ffi_api.SearchPolicyContinueSearchOneRound
        • SketchPolicyNode::ContinueSearchOneRound
          1. SketchPolicyNode::SearchOneRound
          2. SketchPolicyNode::PickStatesWithEpsGreedy
          3. ProgramMeasurerNode::Measure
             1. ProgramMeasurerNode::SilentMeasure(task, input_batch, &result_batch)
                1. ProgramBuilderNode::Build
                   • LocalBuilderNode::Build
                     • auto_scheduler.task_scheduler.measure.local_builder_build
                2. ProgramRunnerNode::Run
                   • LocalRunnerNode::Run
                     • auto_scheduler.task_scheduler.measure.local_run
                       • auto_scheduler.task_scheduler.measure._timed_eval_func
                         1. tvm.runtime.module.load_module(path)
                         2. tvm.runtime.module.Module.time_evaluation
                   • RPCRunnerNode::Run
          4. CostModelNode::Update