| 47 | DEBUG = "[D]" |
| 48 | |
| 49 | def __init__(self): |
| 50 | environ_severity = os.environ.get(self.ENV_VARIABLE) |
| 51 | self._set_from_env = environ_severity is not None |
| 52 | |
| 53 | self.rank: Optional[int] = None |
| 54 | |
| 55 | min_severity = environ_severity.lower() if self._set_from_env else self.DEFAULT_LEVEL |
| 56 | invalid_severity = min_severity not in severity_map |
| 57 | if invalid_severity: |
| 58 | min_severity = self.DEFAULT_LEVEL |
| 59 | |
| 60 | self._min_severity = min_severity |
| 61 | self._trt_logger = trt.Logger(severity_map[min_severity][0]) |
| 62 | self._logger = logging.getLogger(self.PREFIX) |
| 63 | self._logger.propagate = False |
| 64 | handler = logging.StreamHandler(stream=sys.stdout) |
| 65 | handler.setFormatter( |
| 66 | logging.Formatter(fmt="[%(asctime)s] %(message)s", datefmt="%m/%d/%Y-%H:%M:%S") |
| 67 | ) |
| 68 | self._logger.addHandler(handler) |
| 69 | self._logger.setLevel(severity_map[min_severity][1]) |
| 70 | self._polygraphy_logger = G_LOGGER |
| 71 | if self._polygraphy_logger is not None: |
| 72 | self._polygraphy_logger.module_severity = severity_map[min_severity][2] |
| 73 | |
| 74 | # For log_once |
| 75 | self._appeared_keys = set() |
| 76 | |
| 77 | if invalid_severity: |
| 78 | self.warning( |
| 79 | f"Requested log level {environ_severity} is invalid. Using '{self.DEFAULT_LEVEL}' instead" |
| 80 | ) |
| 81 | |
| 82 | def set_rank(self, rank: int): |
| 83 | self.rank = rank |