def check_cache_compile(): files = [ "src/utils/cache_compile.cc", "src/utils/log.cc", "src/utils/tracer.cc", "src/utils/jit_utils.cc", "src/utils/str_utils.cc", ] if os.name == 'nt': files = [x.replace('/', '\\') for x in files] global jit_utils_core_files jit_utils_core_files = files recompile = compile( cc_path, cc_flags + f" {opt_flags} ", files, jit_utils.cache_path + '/jit_utils_core' + extension_suffix, True) if recompile and jit_utils.cc: LOG.e("jit_utils updated, please rerun your command.") sys.exit(0) if not jit_utils.cc: with jit_utils.import_scope(import_flags): jit_utils.try_import_jit_utils_core() assert jit_utils.cc # recompile, generate cache key compile(cc_path, cc_flags + f" {opt_flags} ", files, jit_utils.cache_path + '/jit_utils_core' + extension_suffix, True)
def check_cache_compile(): files = [ "src/utils/cache_compile.cc", "src/utils/log.cc", "src/utils/tracer.cc", "src/utils/jit_utils.cc", ] global jit_utils_core_files jit_utils_core_files = files recompile = compile(cc_path, cc_flags+f" {opt_flags} ", files, 'jit_utils_core'+extension_suffix, True) if recompile and jit_utils.cc: LOG.e("jit_utils updated, please restart jittor.") sys.exit(0) if not jit_utils.cc: with jit_utils.import_scope(import_flags): jit_utils.try_import_jit_utils_core() assert jit_utils.cc # recompile, generate cache key compile(cc_path, cc_flags+f" {opt_flags} ", files, 'jit_utils_core'+extension_suffix, True)
if os.environ.get("debug") == "1": is_debug = 1 global cc_flags cc_flags += " -g -DNODE_MEMCHECK " cc_flags = " " # os.RTLD_NOW | os.RTLD_GLOBAL cause segfault when import torch first import_flags = os.RTLD_NOW | os.RTLD_GLOBAL | os.RTLD_DEEPBIND # if cc_type=="icc": # # weird link problem, icc omp library may conflict and cause segfault # import_flags = os.RTLD_NOW | os.RTLD_GLOBAL dlopen_flags = os.RTLD_NOW | os.RTLD_GLOBAL | os.RTLD_DEEPBIND with jit_utils.import_scope(import_flags): jit_utils.try_import_jit_utils_core() jittor_path = find_jittor_path() check_debug_flags() sys.path.append(cache_path) with jit_utils.import_scope(import_flags): jit_utils.try_import_jit_utils_core() python_path = sys.executable py3_config_paths = [ sys.executable + "-config", os.path.dirname(sys.executable) + f"/python3.{sys.version_info.minor}-config", f"/usr/bin/python3.{sys.version_info.minor}-config",
# *************************************************************** # Copyright (c) 2021 Jittor. All Rights Reserved. # Maintainers: # Guowei Yang <*****@*****.**> # Dun Liang <*****@*****.**>. # # This file is subject to the terms and conditions defined in # file 'LICENSE.txt', which is part of this source code package. # *************************************************************** import sys import os os.environ["log_silent"] = "1" import re import jittor_utils as jit_utils from jittor_utils import LOG jit_utils.try_import_jit_utils_core(silent=True) def my_split(str): res = [] last = -1 for i in range(len(str)): if str[i] == " " or str[i] == "\t": if last < i - 1: res.append(str[last + 1:i]) last = i elif i == len(str) - 1: res.append(str[last + 1:]) return res