fix argument retrieval if subparser is not active

apache · Jan 23, 2024 · 33562e7 · 33562e7
1 parent 5e7f442
commit 33562e7
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/python/tvm/driver/tvmc/compiler.py b/python/tvm/driver/tvmc/compiler.py
@@ -71,9 +71,13 @@ def add_compile_parser(subparsers, main_parser, json_params, argv):
     disposable_parser = TVMCSuppressedArgumentParser(main_parser)
     try:
         known_args, _ = disposable_parser.parse_known_args(argv)
-        _handle_extensions(known_args.experimental_tvm_extension)
     except TVMCException:
         pass
+    try:
+        ext_dirs = known_args.experimental_tvm_extension
+    except AttributeError:
+        ext_dirs = []
+    _handle_extensions(ext_dirs)
 
     parser.add_argument(
         "--cross-compiler",