[tvmc] Introduce 'tune' subcommand (part 3/4) (#6537)
authorLeandro Nunes <leandron85@gmail.com>
Tue, 29 Sep 2020 16:40:46 +0000 (17:40 +0100)
committerGitHub <noreply@github.com>
Tue, 29 Sep 2020 16:40:46 +0000 (09:40 -0700)
commitc0a6bc32d3b6e4ac07ab4d0a678d5189fee5241b
tree611cdb314e8fb213eeea3d1856461efffe33768b
parent27abfadc55e79e3d40b7de7d4b87eb87a19b7b97
[tvmc] Introduce 'tune' subcommand (part 3/4) (#6537)

* tvmc: introduce 'tune' subcommand (part 3/4)

 * introduces a subcommand to drive auto-tuning

Co-authored-by: Matthew Barrett <matthew.barrett@arm.com>
Co-authored-by: Luke Hutton <luke.hutton@arm.com>
Co-authored-by: Giuseppe Rossini <giuseppe.rossini@arm.com>
* [tvmc] address code review comments

* adjust --min-repeat-ms default value logic

* re-arrange rpc arguments to be --rpc-tracker=hostname:port and --rpc-key=str

* use a local reference of the tvmc logger

* add --target-host, default to llvm

Co-authored-by: Matthew Barrett <matthew.barrett@arm.com>
Co-authored-by: Luke Hutton <luke.hutton@arm.com>
Co-authored-by: Giuseppe Rossini <giuseppe.rossini@arm.com>
python/tvm/driver/tvmc/__init__.py
python/tvm/driver/tvmc/autotuner.py [new file with mode: 0644]
python/tvm/driver/tvmc/common.py
tests/python/driver/tvmc/test_autotuner.py [new file with mode: 0644]