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

fix arch90

This commit is contained in:
DongYang Li 2025-02-09 02:30:35 +08:00 committed by GitHub
parent b166e4e385
commit 86841e858d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python/jittor/compiler.py
View File

@ -1371,7 +1371,7 @@ if has_cuda and is_cuda:
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 = 89
max_arch = 90
if nvcc_version < [11,]:
max_arch = 75
elif nvcc_version < [11,1]: