polish cu key

This commit is contained in:
Dun Liang 2021-11-04 20:51:31 +08:00
parent 5695a2e61f
commit 0274c22afd
2 changed files with 2 additions and 1 deletions

View File

@ -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

View File

@ -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)