The goal of this project is to combine Ansor and AutoTVM. We want to use Ansor to do some exploration of kernel configurations, and then use AutoTVM to exploit the best solution found by Ansor.

From Bennu, It's not necessary to build the project. But, It depends on some dependencies from other projects to work that might be necessary to build.

These are the most important dependencies for running Bennu:

Generating the Ansor template with 10k trials for a specific machine architecture.

  python3 benchmarks/mm.py -m ansor -a x86 -l results/x86_mm.json

Optimize Ansor template using Droplet Search with 100 trials:

  python3 benchmarks/mm.py -m droplet -a x86 -l results/x86_mm.json -t 100

Generating the Ansor template with 10k trials for a specific machine architecture.

python3 benchmarks/models_onnx.py -m ansor -a x86 -t 10000 -l results/x86_resnet18_10k.json -b models/resnet18.onnx

Optimize Ansor template using Droplet Search with 100 trials:

python3 benchmarks/models_onnx.py -m droplet -a x86 -t 100 -l results/x86_resnet18_10k.json -b models/resnet18.onnx