From 6069a707605a03cbbf174652b86a6728b0f3b4ab Mon Sep 17 00:00:00 2001
From: Nadav Eidelstein <30617226+nodeav@users.noreply.github.com>
Date: Fri, 14 May 2021 20:07:06 +0300
Subject: [PATCH] Fix minor issues in the tvmc tune CLI (#8039)

* [TVMC] convert timeout flag to int

fixes Check failed: type_code_ == kDLInt (11 vs. 0) : expected int but got str
when setting the timeout option using the cli flag.

* [TVMC] fix typo in tvmc tune help
---
 python/tvm/driver/tvmc/autotuner.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/python/tvm/driver/tvmc/autotuner.py b/python/tvm/driver/tvmc/autotuner.py
index 8f94c53045e5..b92ab86ef621 100644
--- a/python/tvm/driver/tvmc/autotuner.py
+++ b/python/tvm/driver/tvmc/autotuner.py
@@ -116,7 +116,7 @@ def add_tune_parser(subparsers):
         help="the host compilation target, defaults to 'llvm'",
         default="llvm",
     )
-    parser.add_argument("--timeout", default=10, help="compilation timeout, in seconds")
+    parser.add_argument("--timeout", type=int, default=10, help="compilation timeout, in seconds")
     parser.add_argument(
         "--trials",
         type=int,
@@ -142,7 +142,7 @@ def add_tune_parser(subparsers):
 
     auto_scheduler_group = parser.add_argument_group(
         "Autoscheduler options",
-        "Autoscheduler options, used when --enabled-auto-scheduler is provided",
+        "Autoscheduler options, used when --enable-autoscheduler is provided",
     )
 
     auto_scheduler_group.add_argument(