From 2e9c804ad0fa3eebe739e1784d357e5f6b4e89c5 Mon Sep 17 00:00:00 2001 From: kaiyaointel Date: Mon, 5 Dec 2022 15:39:21 +0800 Subject: [PATCH 1/2] update launcher to fit multi-item input --- neural_coder/__main__.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/neural_coder/__main__.py b/neural_coder/__main__.py index 8d9da0472c8..feeaa60cee2 100644 --- a/neural_coder/__main__.py +++ b/neural_coder/__main__.py @@ -25,10 +25,10 @@ def parse_args(): """ parser = ArgumentParser(description="command-launch a Python script with quantization auto-enabled") - parser.add_argument("--opt", type=str, default="", + parser.add_argument("-o", "--opt", type=str, default="", help="optimization feature to enable") - parser.add_argument("--approach", type=str, default="static", + parser.add_argument("-a", "--approach", type=str, default="static", help="quantization approach (strategy)") parser.add_argument('--config', type=str, default="", @@ -54,13 +54,13 @@ def parse_args(): from neural_coder import enable if args.opt == "": if args.approach == "static": - features=["pytorch_inc_static_quant_fx"] + features = ["pytorch_inc_static_quant_fx"] if args.approach == "static_ipex": - features=["pytorch_inc_static_quant_ipex"] + features = ["pytorch_inc_static_quant_ipex"] if args.approach == "dynamic": - features=["pytorch_inc_dynamic_quant"] + features = ["pytorch_inc_dynamic_quant"] else: - features=[args.opt] + features = args.opt.split(",") enable( code=script_copied, features=features, From 68bd1421cd97b09708b2bc15d3e213433712c174 Mon Sep 17 00:00:00 2001 From: kaiyaointel Date: Mon, 5 Dec 2022 15:40:15 +0800 Subject: [PATCH 2/2] Update __main__.py --- neural_coder/__main__.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/neural_coder/__main__.py b/neural_coder/__main__.py index feeaa60cee2..b8bfc27c863 100644 --- a/neural_coder/__main__.py +++ b/neural_coder/__main__.py @@ -61,6 +61,8 @@ def parse_args(): features = ["pytorch_inc_dynamic_quant"] else: features = args.opt.split(",") + +# execute optimization enabling enable( code=script_copied, features=features,