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

polish nvcc_flags

This commit is contained in:
lzhengning 2022-02-18 14:55:00 +08:00
parent e9aca0444c
commit 8c0f66c638
2 changed files with 3 additions and 2 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.38'
__version__ = '1.3.1.39'
from jittor_utils import lock
with lock.lock_scope():
ori_int = int

3
python/jittor/compiler.py
View File

@ -1314,7 +1314,8 @@ with jit_utils.import_scope(import_flags):
flags = core.Flags()
if has_cuda:
nvcc_flags = convert_nvcc_flags(cc_flags)
nvcc_flags = " " + os.environ.get("nvcc_flags", "") + " "
nvcc_flags += convert_nvcc_flags(cc_flags)
nvcc_version = list(jit_utils.get_int_version(nvcc_path))
max_arch = 1000
if nvcc_version < [11,]: