AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Create general_options() func to reduce code in parser

Browse Source 
Signed-off-by: colramos-amd <colramos@amd.com>
This commit is contained in:
colramos-amd
2024-03-22 17:12:54 -05:00
committed by Cole Ramos
parent d72ef52208
commit 482fd6f2ca
1 changed files with 24 additions and 64 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/argparser.py
+24 -64
View File
@@ -33,26 +33,12 @@ def print_avail_arch(avail_arch: list):
ret_str += "\n\t\t {}".format(arch)
return ret_str
def omniarg_parser(parser, omniperf_home, supported_archs, omniperf_version):
# -----------------------------------------
# Parse arguments (dependent on mode)
# -----------------------------------------
## General Command Line Options
## ----------------------------
def add_general_group(parser, omniperf_version):
general_group = parser.add_argument_group("General Options")
parser._positionals.title = "Modes"
parser._optionals.title = "Help"
general_group.add_argument(
"-v", "--version", action="version", version=omniperf_version["ver_pretty"]
)
general_group.add_argument(
"-s", "--specs", action="store_true", help="Print system specs."
)
general_group.add_argument(
"-q", "--quiet", action="store_true", help="Run in quiet mode."
)
general_group.add_argument(
"-V",
"--verbose",
@@ -60,6 +46,25 @@ def omniarg_parser(parser, omniperf_home, supported_archs, omniperf_version):
action="count",
default=0,
)
general_group.add_argument(
"-q", "--quiet", action="store_true", help="Run in quiet mode."
)
# Nowhere to load specs from in db mode
if "database" not in parser.usage:
general_group.add_argument(
"-s", "--specs", action="store_true", help="Print system specs."
)
def omniarg_parser(parser, omniperf_home, supported_archs, omniperf_version):
# -----------------------------------------
# Parse arguments (dependent on mode)
# -----------------------------------------
## General Command Line Options
## ----------------------------
add_general_group(parser, omniperf_version)
parser._positionals.title = "Modes"
parser._optionals.title = "Help"
subparsers = parser.add_subparsers(
dest="mode", help="Select mode of interaction with the target application:"
@@ -91,27 +96,10 @@ Examples:
)
profile_parser._optionals.title = "Help"
general_group = profile_parser.add_argument_group("General Options")
add_general_group(profile_parser, omniperf_version)
profile_group = profile_parser.add_argument_group("Profile Options")
roofline_group = profile_parser.add_argument_group("Standalone Roofline Options")
general_group.add_argument(
"-v", "--version", action="version", version=omniperf_version["ver_pretty"]
)
general_group.add_argument(
"-q", "--quiet", action="store_true", help="Run in quiet mode."
)
general_group.add_argument(
"-V",
"--verbose",
help="Increase output verbosity (use multiple times for higher levels)",
action="count",
default=0,
)
general_group.add_argument(
"-s", "--specs", action="store_true", help="Print system specs."
)
profile_group.add_argument(
"-n",
"--name",
@@ -300,24 +288,10 @@ Examples:
)
db_parser._optionals.title = "Help"
general_group = db_parser.add_argument_group("General Options")
add_general_group(db_parser, omniperf_version)
interaction_group = db_parser.add_argument_group("Interaction Type")
connection_group = db_parser.add_argument_group("Connection Options")
general_group.add_argument(
"-v", "--version", action="version", version=omniperf_version["ver_pretty"]
)
general_group.add_argument(
"-V",
"--verbose",
help="Increase output verbosity (use multiple times for higher levels)",
action="count",
default=0,
)
general_group.add_argument(
"-s", "--specs", action="store_true", help="Print system specs."
)
interaction_group.add_argument(
"-i",
"--import",
@@ -407,24 +381,10 @@ Examples:
)
analyze_parser._optionals.title = "Help"
general_group = analyze_parser.add_argument_group("General Options")
add_general_group(analyze_parser, omniperf_version)
analyze_group = analyze_parser.add_argument_group("Analyze Options")
analyze_advanced_group = analyze_parser.add_argument_group("Advanced Options")
general_group.add_argument(
"-v", "--version", action="version", version=omniperf_version["ver_pretty"]
)
general_group.add_argument(
"-V",
"--verbose",
help="Increase output verbosity (use multiple times for higher levels)",
action="count",
default=0,
)
general_group.add_argument(
"-s", "--specs", action="store_true", help="Print system specs."
)
analyze_group.add_argument(
"-p",
"--path",