Skip to content

Commit 16b2fe7

Browse files
kaikaiyaozehao-intel
authored andcommitted
Neural Coder launcher debug (#215)
* update launcher to fit multi-item input * Update __main__.py Signed-off-by: zehao-intel <[email protected]>
1 parent 7bc4822 commit 16b2fe7

File tree

1 file changed

+9
-6
lines changed

1 file changed

+9
-6
lines changed

neural_coder/__main__.py

Lines changed: 9 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -25,10 +25,11 @@ def parse_args():
2525
"""
2626
parser = ArgumentParser(description="command-launch a Python script with quantization auto-enabled")
2727

28-
parser.add_argument("--opt", type=str, default="",
28+
parser.add_argument("-o", "--opt", type=str, default="",
2929
help="optimization feature to enable")
3030

31-
parser.add_argument("--approach", type=str, default="dynamic",
31+
parser.add_argument("-a", "--approach", type=str, default="static",
32+
3233
help="quantization approach (strategy)")
3334

3435
parser.add_argument('--config', type=str, default="",
@@ -54,13 +55,15 @@ def parse_args():
5455
from neural_coder import enable
5556
if args.opt == "":
5657
if args.approach == "static":
57-
features=["pytorch_inc_static_quant_fx"]
58+
features = ["pytorch_inc_static_quant_fx"]
5859
if args.approach == "static_ipex":
59-
features=["pytorch_inc_static_quant_ipex"]
60+
features = ["pytorch_inc_static_quant_ipex"]
6061
if args.approach == "dynamic":
61-
features=["pytorch_inc_dynamic_quant"]
62+
features = ["pytorch_inc_dynamic_quant"]
6263
else:
63-
features=[args.opt]
64+
features = args.opt.split(",")
65+
66+
# execute optimization enabling
6467
enable(
6568
code=script_copied,
6669
features=features,

0 commit comments

Comments
 (0)