Skip to content

Commit

Permalink
[TVMC] 'tvmc tune' --rpc-tracker and --rpc-tracker fail due to argpar…
Browse files Browse the repository at this point in the history
…se misconfiguration (apache#6822)

Fix an error with `tvmc tune`, that causes --rpc-tracker and --rpc-key to be identified as a list of strings, rather than the expected string type.

Removing the unnecessary nargs solves the issues.

This is a follow-up of apache#6762
  • Loading branch information
Giuseppe Rossini authored and Trevor Morris committed Dec 2, 2020
1 parent d268c8e commit 5f5c7ad
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions python/tvm/driver/tvmc/autotuner.py
Original file line number Diff line number Diff line change
Expand Up @@ -92,12 +92,10 @@ def add_tune_parser(subparsers):
)
parser.add_argument(
"--rpc-key",
nargs=1,
help="the RPC tracker key of the target device. Required when --rpc-tracker is provided.",
)
parser.add_argument(
"--rpc-tracker",
nargs=1,
help="hostname (required) and port (optional, defaults to 9090) of the RPC tracker, "
"e.g. '192.168.0.100:9999'",
)
Expand Down

0 comments on commit 5f5c7ad

Please sign in to comment.