diff options
author | Christopher Christofi <77968333+ChristopherChristofi@users.noreply.github.com> | 2023-07-22 17:27:49 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-07-22 12:27:49 -0400 |
commit | c811b71336175480d29c4763681ace5c0be3c335 (patch) | |
tree | 35dc3352981312573cd5f1968532b3b2d8ffc99e | |
parent | 366c798b87783bbd9db7c74fd96fc5228d7b47b4 (diff) | |
download | spack-c811b71336175480d29c4763681ace5c0be3c335.tar.gz spack-c811b71336175480d29c4763681ace5c0be3c335.tar.bz2 spack-c811b71336175480d29c4763681ace5c0be3c335.tar.xz spack-c811b71336175480d29c4763681ace5c0be3c335.zip |
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 <ajstewart426@gmail.com>
* 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 <ajstewart426@gmail.com>
---------
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
-rw-r--r-- | var/spack/repos/builtin/packages/py-jaxlib/package.py | 7 |
1 files changed, 7 insertions, 0 deletions
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") |