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

lower arch support for 30

This commit is contained in:
Dun Liang 2021-10-01 19:56:47 +08:00
parent 0fa0584fd3
commit 6300b0908f
2 changed files with 2 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. # file 'LICENSE.txt', which is part of this source code package.
# *************************************************************** # ***************************************************************
__version__ = '1.3.0.2' __version__ = '1.3.0.3'
from jittor_utils import lock from jittor_utils import lock
with lock.lock_scope(): with lock.lock_scope():
ori_int = int ori_int = int

2
python/jittor/compiler.py
View File

@ -1283,7 +1283,7 @@ if has_cuda:
archs = [] archs = []
for arch in flags.cuda_archs: for arch in flags.cuda_archs:
if arch<50: if arch<50:
LOG.w(f"CUDA arch({arch})<50 is not supported") LOG.w(f"CUDA arch({arch})<30 is not supported")
continue continue
archs.append(arch) archs.append(arch)
nvcc_flags += f" -arch=compute_{min(flags.cuda_archs)} " nvcc_flags += f" -arch=compute_{min(flags.cuda_archs)} "