diff --git a/python/jittor/compiler.py b/python/jittor/compiler.py index f1e2f2a0..a979e9ec 100644 --- a/python/jittor/compiler.py +++ b/python/jittor/compiler.py @@ -1246,7 +1246,11 @@ jit_utils.add_backend(corex_compiler) for mod in jit_utils.backends: if mod.check(): 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 gen_jit_flags()