Neural Coder enable launcher bench (#260)

* enable auto bench in launcher

* debug

* debug

Author: kaikaiyao

neural_coder/__main__.py

@@ -35,6 +35,9 @@ def parse_args():
     parser.add_argument('--config', type=str, default="",
                         help='quantization configuration file path')
 
+    parser.add_argument('-b', '--bench', default=False, action='store_true',
+                        help='conduct auto_quant benchmark instead of enable')
+
     # positional
     parser.add_argument("script", type=str,
                         help="The full path to the script to be launched. "
@@ -51,34 +54,41 @@ def parse_args():
 script_copied = args.script[:-3] + "_optimized.py"
 shutil.copy(args.script, script_copied)
 
-# optimize on copied script with Neural Coder
-from neural_coder import enable
-if args.opt == "":
-    if args.approach == "static":
-        features = ["pytorch_inc_static_quant_fx"]
-    if args.approach == "static_ipex":
-        features = ["pytorch_inc_static_quant_ipex"]
-    if args.approach == "dynamic":
-        features = ["pytorch_inc_dynamic_quant"]
-else:
-    features = args.opt.split(",")
-
-# execute optimization enabling
-enable(
-    code=script_copied,
-    features=features,
-    overwrite=True,
-)
-
-# execute on copied script, which has already been optimized
-cmd = []
-
-cmd.append(sys.executable) # "/xxx/xxx/python"
-cmd.append("-u")
-cmd.append(script_copied)
-cmd.extend(args.script_args)
-
-cmd = " ".join(cmd) # list convert to string
-
-process = subprocess.Popen(cmd, env=os.environ, shell=True)  # nosec
-process.wait()
+if not args.bench: # enable
+    # optimize on copied script with Neural Coder
+    from neural_coder import enable
+    if args.opt == "":
+        if args.approach == "static":
+            features = ["pytorch_inc_static_quant_fx"]
+        if args.approach == "static_ipex":
+            features = ["pytorch_inc_static_quant_ipex"]
+        if args.approach == "dynamic":
+            features = ["pytorch_inc_dynamic_quant"]
+    else:
+        features = args.opt.split(",")
+
+    # execute optimization enabling
+    enable(
+        code=script_copied,
+        features=features,
+        overwrite=True,
+    )
+
+    # execute on copied script, which has already been optimized
+    cmd = []
+
+    cmd.append(sys.executable) # "/xxx/xxx/python"
+    cmd.append("-u")
+    cmd.append(script_copied)
+    cmd.extend(args.script_args)
+
+    cmd = " ".join(cmd) # list convert to string
+
+    process = subprocess.Popen(cmd, env=os.environ, shell=True)  # nosec
+    process.wait()
+else: # auto_quant
+    from neural_coder import auto_quant
+    auto_quant(
+        code=script_copied,
+        args=' '.join(args.script_args), # convert list of strings to a single string
+    )