[tvm] branch main updated: Fix incorrect device name in TVMC. (#8181)

[ma...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

masahi pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/main by this push:
     new 072a3d2  Fix incorrect device name in TVMC. (#8181)
072a3d2 is described below

commit 072a3d236b10708a2f419727917ed85ee0941d5d
Author: Matt Welsh (OctoML) <63...@users.noreply.github.com>
AuthorDate: Sun Jun 6 12:50:58 2021 -0700

    Fix incorrect device name in TVMC. (#8181)
    
    * Fix incorrect device name in TVMC.
    
    * Rename gpu -> cuda.
    
    * Bump CI.
---
 python/tvm/driver/tvmc/runner.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/python/tvm/driver/tvmc/runner.py b/python/tvm/driver/tvmc/runner.py
index ba0f7d2..191f861 100644
--- a/python/tvm/driver/tvmc/runner.py
+++ b/python/tvm/driver/tvmc/runner.py
@@ -51,7 +51,7 @@ def add_run_parser(subparsers):
     #      like 'webgpu', etc (@leandron)
     parser.add_argument(
         "--device",
-        choices=["cpu", "gpu", "cl"],
+        choices=["cpu", "cuda", "cl"],
         default="cpu",
         help="target device to run the compiled module. Defaults to 'cpu'",
     )
@@ -323,7 +323,7 @@ def run_module(
     tvmc_package: TVMCPackage
         The compiled model package object that will be run.
     device: str,
-        the device (e.g. "cpu" or "gpu") to be targeted by the RPC
+        the device (e.g. "cpu" or "cuda") to be targeted by the RPC
         session, local or remote).
     hostname : str, optional
         The hostname of the target device on which to run.