def try_import_jit_utils_core(silent=None): global cc if cc: return if not (silent is None): prev = os.environ.get("log_silent", "0") os.environ["log_silent"] = str(int(silent)) try: # if is in notebook, must log sync, and we redirect the log if is_in_ipynb: os.environ["log_sync"] = "1" import jit_utils_core as cc if is_in_ipynb: cc.ostream_redirect(True, True) except Exception as _: pass if not (silent is None): os.environ["log_silent"] = prev
def try_import_jit_utils_core(silent=None): global cc if cc: return if not (silent is None): prev = os.environ.get("log_silent", "0") os.environ["log_silent"] = str(int(silent)) try: # if is in notebook, must log sync, and we redirect the log if is_in_ipynb: os.environ["log_sync"] = "1" # jittor\src\utils\jit_utils.cc PYBIND11_MODULE(jit_utils_core, m) # pybind 调用 c++源码 实现 # jittor\src\utils\log.cc import jit_utils_core as cc if is_in_ipynb: global redirector redirector = cc.ostream_redirect(stdout=True, stderr=True) redirector.__enter__() except Exception as _: pass if not (silent is None): os.environ["log_silent"] = prev