lzhengning
/
JittorMirror
mirror of https://github.com/Jittor/Jittor
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add nvcc_path check

This commit is contained in:
Dun Liang 2022-01-18 16:32:23 +08:00
parent 6a372f5f4f
commit 5d191e6247
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python/jittor/__init__.py
View File

@ -9,7 +9,7 @@
# file 'LICENSE.txt', which is part of this source code package.
# ***************************************************************
__version__ = '1.3.1.37'
__version__ = '1.3.1.38'
from jittor_utils import lock
with lock.lock_scope():
ori_int = int

2
python/jittor/compiler.py
View File

@ -974,6 +974,8 @@ if not nvcc_path:
nvcc_path = try_find_exe(nvcc_path)
if nvcc_path is None:
nvcc_path = ""
if "nvcc_path" in os.environ:
nvcc_path = os.environ["nvcc_path"]
gdb_path = env_or_try_find('gdb_path', 'gdb')
addr2line_path = try_find_exe('addr2line')
has_pybt = check_pybt(gdb_path, python_path)