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

update cutt

This commit is contained in:
Dun Liang 2020-12-21 15:43:42 +08:00
parent 8ed660bf23
commit 10ea7eab35
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
python/jittor/compile_extern.py
View File

@ -183,7 +183,7 @@ def install_cutt(root_folder):
filename = "cutt-master.zip"
fullname = os.path.join(root_folder, filename)
dirname = os.path.join(root_folder, filename.replace(".zip",""))
true_md5 = "a6f4f7f75310a69b131e21f1ebec768a"
true_md5 = "af5bc35eea1832a42c0e0011659b7209"
if os.path.exists(fullname):
md5 = run_cmd('md5sum '+fullname).split()[0]
@ -205,7 +205,11 @@ def install_cutt(root_folder):
zf.close()
LOG.i("installing cutt...")
run_cmd(f"make", cwd=dirname)
arch_flag = ""
if len(flags.cuda_archs):
arch_flag = f" -arch=compute_{min(flags.cuda_archs)} "
arch_flag += ''.join(map(lambda x:f' -code=sm_{x} ', flags.cuda_archs))
run_cmd(f"make NVCC_GENCODE='{arch_flag}'", cwd=dirname)
return dirname
def setup_cutt():