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

fix windows cuda.

This commit is contained in:
Exusial 2022-12-25 11:13:22 +08:00 committed by Zheng-Ning Liu
parent 6b7d1a27d4
commit 35763e6c2c
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
python/jittor/compiler.py
View File

@ -1246,7 +1246,11 @@ jit_utils.add_backend(corex_compiler)
for mod in jit_utils.backends: for mod in jit_utils.backends:
if mod.check(): if mod.check():
break break
is_cuda = os.path.basename(nvcc_path) == "nvcc"
if not os.name == 'nt':
is_cuda = os.path.basename(nvcc_path) == "nvcc"
else:
is_cuda = os.path.basename(nvcc_path) == "nvcc.exe"
# build core # build core
gen_jit_flags() gen_jit_flags()