bump version, merge pull request #16 from AMYPAD/fix-multi-arch

AMYPAD · Sep 25, 2021 · f228d8d · f228d8d
2 parents d7d262c + f844769
commit f228d8d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -21,7 +21,7 @@
     nvcc_arch_raw = map(cuinfo.compute_capability, range(cuinfo.num_devices()))
     nvcc_arches = {"%d%d" % i for i in nvcc_arch_raw if i >= (3, 5)}
     if nvcc_arches:
-        cmake_args.append("-DCMAKE_CUDA_ARCHITECTURES=" + " ".join(sorted(nvcc_arches)))
+        cmake_args.append("-DCMAKE_CUDA_ARCHITECTURES=" + ";".join(sorted(nvcc_arches)))
 except Exception as exc:
     log.warning("Import or CUDA device detection error:\n%s", exc)