From 6de15707c1f9ec7409432aea231e149c91b79626 Mon Sep 17 00:00:00 2001 From: gxiaotian Date: Tue, 30 Jun 2020 10:55:19 +0800 Subject: [PATCH] Add OpEvo example (#2549) --- README.md | 3 +- docs/en_US/TrialExample/OpEvoExamples.md | 85 ++++ docs/en_US/examples.rst | 1 + examples/trials/systems/opevo/Dockerfile | 42 ++ examples/trials/systems/opevo/Makefile | 14 + .../systems/opevo/src/algorithms/gbfs.py | 278 +++++++++++ .../systems/opevo/src/algorithms/na2c.py | 401 +++++++++++++++ .../systems/opevo/src/algorithms/opevo.py | 455 ++++++++++++++++++ .../opevo/src/compiler_auto_tune_stable.py | 399 +++++++++++++++ .../bmm/B960N128K128M64PNN/config_opevo.yml | 25 + .../bmm/B960N128K128M64PNN/search_space.json | 1 + .../bmm/B960N128K128M64PTN/config_opevo.yml | 25 + .../bmm/B960N128K128M64PTN/search_space.json | 1 + .../bmm/B960N128K64M128PNT/config_opevo.yml | 25 + .../bmm/B960N128K64M128PNT/search_space.json | 1 + .../N512C3HW227F64K11ST4PD0/config_opevo.yml | 25 + .../N512C3HW227F64K11ST4PD0/search_space.json | 1 + .../N512C64HW27F192K5ST1PD2/config_opevo.yml | 25 + .../N512C64HW27F192K5ST1PD2/search_space.json | 1 + .../mm/N512K1024M1024/config_gbfs.yml | 23 + .../mm/N512K1024M1024/config_na2c.yml | 22 + .../mm/N512K1024M1024/config_opevo.yml | 25 + .../mm/N512K1024M1024/search_space.json | 1 + .../mm/N512K1024M4096/config_gbfs.yml | 23 + .../mm/N512K1024M4096/config_na2c.yml | 22 + .../mm/N512K1024M4096/config_opevo.yml | 25 + .../mm/N512K1024M4096/search_space.json | 1 + .../mm/N512K4096M1024/config_gbfs.yml | 23 + .../mm/N512K4096M1024/config_na2c.yml | 22 + .../mm/N512K4096M1024/config_opevo.yml | 25 + .../mm/N512K4096M1024/search_space.json | 1 + examples/trials/systems/opevo/src/run.sh | 25 + .../opevo/src/templates/batch_matmul.py | 119 +++++ .../opevo/src/templates/convfwd_direct.py | 130 +++++ .../systems/opevo/src/templates/matmul.py | 111 +++++ .../systems/opevo/tvm_patches/libcuda.so.1 | Bin 0 -> 46448 bytes .../systems/opevo/tvm_patches/tvm_v0.6.patch | 84 ++++ 37 files changed, 2489 insertions(+), 1 deletion(-) create mode 100644 docs/en_US/TrialExample/OpEvoExamples.md create mode 100644 examples/trials/systems/opevo/Dockerfile create mode 100644 examples/trials/systems/opevo/Makefile create mode 100644 examples/trials/systems/opevo/src/algorithms/gbfs.py create mode 100644 examples/trials/systems/opevo/src/algorithms/na2c.py create mode 100644 examples/trials/systems/opevo/src/algorithms/opevo.py create mode 100644 examples/trials/systems/opevo/src/compiler_auto_tune_stable.py create mode 100644 examples/trials/systems/opevo/src/experiments/bmm/B960N128K128M64PNN/config_opevo.yml create mode 100644 examples/trials/systems/opevo/src/experiments/bmm/B960N128K128M64PNN/search_space.json create mode 100644 examples/trials/systems/opevo/src/experiments/bmm/B960N128K128M64PTN/config_opevo.yml create mode 100644 examples/trials/systems/opevo/src/experiments/bmm/B960N128K128M64PTN/search_space.json create mode 100644 examples/trials/systems/opevo/src/experiments/bmm/B960N128K64M128PNT/config_opevo.yml create mode 100644 examples/trials/systems/opevo/src/experiments/bmm/B960N128K64M128PNT/search_space.json create mode 100644 examples/trials/systems/opevo/src/experiments/conv/N512C3HW227F64K11ST4PD0/config_opevo.yml create mode 100644 examples/trials/systems/opevo/src/experiments/conv/N512C3HW227F64K11ST4PD0/search_space.json create mode 100644 examples/trials/systems/opevo/src/experiments/conv/N512C64HW27F192K5ST1PD2/config_opevo.yml create mode 100644 examples/trials/systems/opevo/src/experiments/conv/N512C64HW27F192K5ST1PD2/search_space.json create mode 100644 examples/trials/systems/opevo/src/experiments/mm/N512K1024M1024/config_gbfs.yml create mode 100644 examples/trials/systems/opevo/src/experiments/mm/N512K1024M1024/config_na2c.yml create mode 100644 examples/trials/systems/opevo/src/experiments/mm/N512K1024M1024/config_opevo.yml create mode 100644 examples/trials/systems/opevo/src/experiments/mm/N512K1024M1024/search_space.json create mode 100644 examples/trials/systems/opevo/src/experiments/mm/N512K1024M4096/config_gbfs.yml create mode 100644 examples/trials/systems/opevo/src/experiments/mm/N512K1024M4096/config_na2c.yml create mode 100644 examples/trials/systems/opevo/src/experiments/mm/N512K1024M4096/config_opevo.yml create mode 100644 examples/trials/systems/opevo/src/experiments/mm/N512K1024M4096/search_space.json create mode 100644 examples/trials/systems/opevo/src/experiments/mm/N512K4096M1024/config_gbfs.yml create mode 100644 examples/trials/systems/opevo/src/experiments/mm/N512K4096M1024/config_na2c.yml create mode 100644 examples/trials/systems/opevo/src/experiments/mm/N512K4096M1024/config_opevo.yml create mode 100644 examples/trials/systems/opevo/src/experiments/mm/N512K4096M1024/search_space.json create mode 100644 examples/trials/systems/opevo/src/run.sh create mode 100644 examples/trials/systems/opevo/src/templates/batch_matmul.py create mode 100644 examples/trials/systems/opevo/src/templates/convfwd_direct.py create mode 100644 examples/trials/systems/opevo/src/templates/matmul.py create mode 100644 examples/trials/systems/opevo/tvm_patches/libcuda.so.1 create mode 100644 examples/trials/systems/opevo/tvm_patches/tvm_v0.6.patch diff --git a/README.md b/README.md index 9a72dae882..3ea89ebdee 100644 --- a/README.md +++ b/README.md @@ -90,6 +90,7 @@ Within the following table, we summarized the current NNI capabilities, we are g
  • Cifar10-pytorch
  • Scikit-learn
  • EfficientNet
  • +
  • Kernel Tunning
  • More...
    @@ -170,7 +171,7 @@ Within the following table, we summarized the current NNI capabilities, we are g
  • Kubeflow
  • FrameworkController on K8S (AKS etc.)
  • -