Skip to content

Commit

Permalink
[tvmc] Introduce 'run' subcommand (part 4/4) (apache#6578)
Browse files Browse the repository at this point in the history
* [tvmc] Introduce 'run' subcommand (part 4/4)

 * Add 'tvmc run' subcommand to execute compiled modules
 * Include options to locally or remotelly using RPC
 * Include support to cpu and gpu devices


Co-authored-by: Marcus Shawcroft <marcus.shawcroft@arm.com>
Co-authored-by: Matthew Barrett <matthew.barrett@arm.com>

* adjust based on code review comments

* make test fixture to safely skip environments without tflite

* make --help option more clear

* improve error message to show expected inputs

* code-review adjusts

* update doc-string to default zeros->random

Co-authored-by: Marcus Shawcroft <marcus.shawcroft@arm.com>
Co-authored-by: Matthew Barrett <matthew.barrett@arm.com>
  • Loading branch information
3 people authored and Tushar Dey committed Oct 15, 2020
1 parent 36626d8 commit 5aa75a2
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions tests/python/driver/tvmc/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -138,7 +138,11 @@ def imagenet_cat(tmpdir_factory):
tmpdir_name = tmpdir_factory.mktemp("data")
cat_file_name = "imagenet_cat.npz"

<<<<<<< HEAD
cat_url = "https://github.com/dmlc/mxnet.js/blob/main/data/cat.png?raw=true"
=======
cat_url = "https://github.com/dmlc/mxnet.js/blob/master/data/cat.png?raw=true"
>>>>>>> [tvmc] Introduce 'run' subcommand (part 4/4) (#6578)
image_path = download_testdata(cat_url, "inputs", module=["tvmc"])
resized_image = Image.open(image_path).resize((224, 224))
image_data = np.asarray(resized_image).astype("float32")
Expand Down

0 comments on commit 5aa75a2

Please sign in to comment.