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

Connect verbosity flag to loglevel

Browse Source 
Signed-off-by: colramos-amd <colramos@amd.com>


[ROCm/rocprofiler-compute commit: 5ed9a5c902]
This commit is contained in:
colramos-amd
2024-02-08 12:23:20 -06:00
committed by Karl W. Schulz
parent a1371462ba
commit 569dff3627
4 changed files with 15 additions and 85 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rocprofiler-compute/src/utils/logger.py
+5 -3
View File
@@ -50,15 +50,17 @@ class ColoredFormatter(logging.Formatter):
return logging.Formatter.format(self, record)
# Setup logger
def setup_logging():
def setup_logging(verbose):
# 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)
# demonstrate override of default loglevel via env variable
loglevel=logging.INFO
# default: set loglevel based on selected verbosity
levels = [logging.INFO, logging.DEBUG, logging.TRACE]
loglevel = levels[min(verbose, len(levels) - 1)] # cap to last level index
# optional: override of default loglevel via env variable
if "OMNIPERF_LOGLEVEL" in os.environ.keys():
loglevel = os.environ['OMNIPERF_LOGLEVEL']
if loglevel in {"DEBUG","debug"}: