TVM

Bayesian Code Diffusion

Posted by Treaseven on August 8, 2025

python3 our.py –target=cuda –model=resnet-18 –log_dir=log_ansor –group_type=sketch –num_measures_per_round=64 –test_idx=0 –num-trials=200

修改的文件 /tvm/auto_scheduler/search_policy.py /tvm/auto_scheduler/task_scheduler.py

/src/auto_scheduler/search_policy/sketch_policy_rules.cc —- InitFillTileSize::Apply /src/auto_scheduler/search_policy/sketch_policy.cc —- SketchPolicyNode::ContinueSearchOneRound、SketchPolicy::SketchPolicy /src/auto_scheduler/search_policy/sketch_policy.h —- SketchPolicyNode SketchPolicy