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

add cutt reference

This commit is contained in:
guoye 2020-03-20 14:29:54 +08:00
parent 4b4ea7bf20
commit eb2b0f7098
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
python/jittor/compile_extern.py
View File

@ -162,6 +162,7 @@ def setup_cuda_lib(lib_name, link=True, extra_flags=""):
LOG.vv(f"Get {lib_name}_ops: "+str(dir(culib_ops)))
def install_cutt(root_folder):
# Modified from: https://github.com/ap-hynninen/cutt
url = "https://cloud.tsinghua.edu.cn/f/4be7e1dd51c6459aa119/?dl=1"
filename = "cutt.tgz"
fullname = os.path.join(root_folder, filename)
@ -176,7 +177,7 @@ def install_cutt(root_folder):
os.system('rm ' + fullname)
os.system('rm -rf ' + dirname)
if not os.path.isfile(os.path.join(dirname, "bin", "cutt_test")):
LOG.i("Downloading cub...")
LOG.i("Downloading cutt...")
download_url_to_local(url, filename, root_folder, true_md5)
import tarfile