microsoft/mscclpp
1
0
Fork 0
mirror of https://github.com/microsoft/mscclpp.git synced 2026-05-13 01:36:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

extract level

Browse Source
This commit is contained in:
Crutcher Dunnavant
2023-04-04 18:29:45 +00:00
parent 0df50830e1
commit aaf93c858d
1 changed files with 13 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
python/src/mscclpp/__init__.py
View File

@@ -1,4 +1,5 @@
import os
import re
import atexit
from typing import Any
import json
@@ -19,9 +20,19 @@ _Comm = _py_mscclpp._Comm
MscclppUniqueId = _py_mscclpp.MscclppUniqueId
MSCCLPP_UNIQUE_ID_BYTES = _py_mscclpp.MSCCLPP_UNIQUE_ID_BYTES
def _setup_logging(level='INFO'):
def _mscclpp_log_cb(msg: str) -> None:
logging.info
level = logging.INFO
if match := re.search(r'MSCCLPP (\w+)', msg):
level = logging._nameToLevel.get(match.group(1), logging.INFO)
logger.log(level, msg)
def _set_mscclpp_debug_log_level(level: str ='INFO'):
os.environ['MSCCLPP_DEBUG'] = level
_py_mscclpp._bind_log_handler(logger.info)
def _setup_logging(level='INFO'):
_set_mscclpp_debug_log_level(level)
_py_mscclpp._bind_log_handler(_mscclpp_log_cb)
# needed to prevent a segfault at exit.
atexit.register(_py_mscclpp._release_log_handler)