polish cu key

2021-11-04 20:51:58 +08:00 · 2021-11-04 20:51:58 +08:00 · b3ee4a54e4
parent 0274c22afd
commit b3ee4a54e4
1 changed files with 1 additions and 1 deletions
--- a/python/jittor/compiler.py
+++ b/python/jittor/compiler.py
@ -991,7 +991,7 @@ if nvcc_path:
            cu += "_sm_" + "_".join(s)
            if "cuda_arch" not in os.environ:
                os.environ["cuda_arch"] = " ".join(cu)
-            cu = cu.replace(":", "")
+            cu = cu.replace(":", "").replace(" ", "")
    except:
        pass
    LOG.i("cuda key:", cu)