From c811b71336175480d29c4763681ace5c0be3c335 Mon Sep 17 00:00:00 2001 From: Christopher Christofi <77968333+ChristopherChristofi@users.noreply.github.com> Date: Sat, 22 Jul 2023 17:27:49 +0100 Subject: py-jaxlib: add conflict for missing cuda cuda_arch value (#39054) * py-jaxlib: add conflict for missing cuda cuda_arch specification * Update var/spack/repos/builtin/packages/py-jaxlib/package.py Co-authored-by: Adam J. Stewart * py-jaxlib: conflict missing cuda_arch value when with cuda * Update var/spack/repos/builtin/packages/py-jaxlib/package.py Co-authored-by: Adam J. Stewart --------- Co-authored-by: Adam J. Stewart --- var/spack/repos/builtin/packages/py-jaxlib/package.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/var/spack/repos/builtin/packages/py-jaxlib/package.py b/var/spack/repos/builtin/packages/py-jaxlib/package.py index d8def42008..b4c7daba27 100644 --- a/var/spack/repos/builtin/packages/py-jaxlib/package.py +++ b/var/spack/repos/builtin/packages/py-jaxlib/package.py @@ -47,6 +47,13 @@ class PyJaxlib(PythonPackage, CudaPackage): depends_on("py-absl-py", when="@:0.3", type=("build", "run")) depends_on("py-flatbuffers@1.12:2", when="@0.1", type=("build", "run")) + conflicts( + "cuda_arch=none", + when="+cuda", + msg="Must specify CUDA compute capabilities of your GPU, see " + "https://developer.nvidia.com/cuda-gpus", + ) + def patch(self): self.tmp_path = tempfile.mkdtemp(prefix="spack") self.buildtmp = tempfile.mkdtemp(prefix="spack") -- cgit v1.2.3-60-g2f50