From e1470d7a389fdfddfe6229387bafd2876a8ff1d6 Mon Sep 17 00:00:00 2001 From: coleramos425 Date: Fri, 29 Mar 2024 12:19:24 -0500 Subject: [PATCH] Register TRACE loglevel ahead of parse_args to avoid error in --specs printout Signed-off-by: coleramos425 [ROCm/rocprofiler-compute commit: 706afa7f37b07b9e677026debec9e8f0fc6a4ecb] --- projects/rocprofiler-compute/src/utils/logger.py | 11 ++++++----- projects/rocprofiler-compute/src/utils/specs.py | 6 ++++-- 2 files changed, 10 insertions(+), 7 deletions(-) diff --git a/projects/rocprofiler-compute/src/utils/logger.py b/projects/rocprofiler-compute/src/utils/logger.py index 05bec4c18b..b84ec477fa 100644 --- a/projects/rocprofiler-compute/src/utils/logger.py +++ b/projects/rocprofiler-compute/src/utils/logger.py @@ -64,7 +64,13 @@ class PlainFormatter(logging.Formatter): # Setup console handler - provided as separate function to be called # prior to argument parsing def setup_console_handler(): + # register a trace level logger + logging.TRACE = logging.DEBUG - 5 + logging.addLevelName(logging.TRACE, "TRACE") + setattr(logging, "TRACE", logging.TRACE) + setattr(logging, "trace", trace_logger) + # setup log formatting color_setting = 0 if "OMNIPERF_COLOR" in os.environ.keys(): color_setting = int(os.environ["OMNIPERF_COLOR"]) @@ -97,11 +103,6 @@ def setup_file_handler(loglevel, workload_dir): # Setup logger priority - called after argument parsing def setup_logging_priority(verbosity, quietmode, appmode): - # register a trace level logger - logging.TRACE = logging.DEBUG - 5 - logging.addLevelName(logging.TRACE, "TRACE") - setattr(logging, "TRACE", logging.TRACE) - setattr(logging, "trace", trace_logger) # set loglevel based on selected verbosity and quietmode levels = [logging.INFO, logging.DEBUG, logging.TRACE] diff --git a/projects/rocprofiler-compute/src/utils/specs.py b/projects/rocprofiler-compute/src/utils/specs.py index 04c0fd7125..a528adc7a9 100644 --- a/projects/rocprofiler-compute/src/utils/specs.py +++ b/projects/rocprofiler-compute/src/utils/specs.py @@ -353,15 +353,17 @@ class MachineSpecs: gpu_l1: str = field( default=None, metadata={ - "doc": "The size of the vL1D cache (per compute-unit) on the accelerators/GPUs in the system in KiB", + "doc": "The size of the vL1D cache (per compute-unit) on the accelerators/GPUs.", "name": "GPU L1", + "unit": "KiB", }, ) gpu_l2: str = field( default=None, metadata={ - "doc": "The size of the vL1D cache (per compute-unit) on the accelerators/GPUs in the system in KiB", + "doc": "The size of the vL1D cache (per compute-unit) on the accelerators/GPUs.", "name": "GPU L2", + "unit": "KiB", }, ) cu_per_gpu: str = field(