mirror of https://github.com/Jittor/Jittor
polish cu key
This commit is contained in:
parent
5695a2e61f
commit
0274c22afd
|
@ -9,7 +9,7 @@
|
|||
# file 'LICENSE.txt', which is part of this source code package.
|
||||
# ***************************************************************
|
||||
|
||||
__version__ = '1.3.1.17'
|
||||
__version__ = '1.3.1.18'
|
||||
from jittor_utils import lock
|
||||
with lock.lock_scope():
|
||||
ori_int = int
|
||||
|
|
|
@ -991,6 +991,7 @@ if nvcc_path:
|
|||
cu += "_sm_" + "_".join(s)
|
||||
if "cuda_arch" not in os.environ:
|
||||
os.environ["cuda_arch"] = " ".join(cu)
|
||||
cu = cu.replace(":", "")
|
||||
except:
|
||||
pass
|
||||
LOG.i("cuda key:", cu)
|
||||
|
|
Loading…
Reference in New Issue