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
