From 35763e6c2cba219514b049e8973c7efb4d8dbd9a Mon Sep 17 00:00:00 2001 From: Exusial Date: Sun, 25 Dec 2022 11:13:22 +0800 Subject: [PATCH] fix windows cuda. --- python/jittor/compiler.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) 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()