summaryrefslogtreecommitdiff
path: root/share/spack
diff options
context:
space:
mode:
authorAdam J. Stewart <ajstewart426@gmail.com>2023-12-01 19:23:58 +0100
committerGitHub <noreply@github.com>2023-12-01 18:23:58 +0000
commitab50aa61db858ca1d5c86da8a45e8eb499ad4d19 (patch)
treea29d96216bc703299f0a756415ee27616a6af85d /share/spack
parentd06a102e69c3b546b50e5dd4290aae9654516ae6 (diff)
downloadspack-ab50aa61db858ca1d5c86da8a45e8eb499ad4d19.tar.gz
spack-ab50aa61db858ca1d5c86da8a45e8eb499ad4d19.tar.bz2
spack-ab50aa61db858ca1d5c86da8a45e8eb499ad4d19.tar.xz
spack-ab50aa61db858ca1d5c86da8a45e8eb499ad4d19.zip
py-keras: add v3.0.0 (#41356)
* py-keras: add v3.0.0 * Older keras actually requires protobuf * Correct url_for_version * Capitalization is important * Keep pil and pydot deps
Diffstat (limited to 'share/spack')
-rw-r--r--share/spack/gitlab/cloud_pipelines/stacks/ml-darwin-aarch64-mps/spack.yaml4
-rw-r--r--share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cpu/spack.yaml4
-rw-r--r--share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cuda/spack.yaml4
-rw-r--r--share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-rocm/spack.yaml4
4 files changed, 12 insertions, 4 deletions
diff --git a/share/spack/gitlab/cloud_pipelines/stacks/ml-darwin-aarch64-mps/spack.yaml b/share/spack/gitlab/cloud_pipelines/stacks/ml-darwin-aarch64-mps/spack.yaml
index 0905305113..0526a4a12f 100644
--- a/share/spack/gitlab/cloud_pipelines/stacks/ml-darwin-aarch64-mps/spack.yaml
+++ b/share/spack/gitlab/cloud_pipelines/stacks/ml-darwin-aarch64-mps/spack.yaml
@@ -24,7 +24,9 @@ spack:
# Keras
# Bazel codesign issues
- # - py-keras
+ # - py-keras backend=tensorflow
+ # - py-keras backend=jax
+ - py-keras backend=torch
- py-keras-applications
- py-keras-preprocessing
- py-keras2onnx
diff --git a/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cpu/spack.yaml b/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cpu/spack.yaml
index 71670d5a91..fdce7f9bb4 100644
--- a/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cpu/spack.yaml
+++ b/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cpu/spack.yaml
@@ -19,7 +19,9 @@ spack:
- py-jaxlib
# Keras
- - py-keras
+ # - py-keras backend=tensorflow
+ - py-keras backend=jax
+ - py-keras backend=torch
- py-keras-applications
- py-keras-preprocessing
- py-keras2onnx
diff --git a/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cuda/spack.yaml b/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cuda/spack.yaml
index 8829169038..1a64d83811 100644
--- a/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cuda/spack.yaml
+++ b/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cuda/spack.yaml
@@ -22,7 +22,9 @@ spack:
- py-jaxlib
# Keras
- - py-keras
+ # - py-keras backend=tensorflow
+ - py-keras backend=jax
+ - py-keras backend=torch
- py-keras-applications
- py-keras-preprocessing
- py-keras2onnx
diff --git a/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-rocm/spack.yaml b/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-rocm/spack.yaml
index 620a95715b..5633d9846a 100644
--- a/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-rocm/spack.yaml
+++ b/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-rocm/spack.yaml
@@ -24,7 +24,9 @@ spack:
- py-jaxlib
# Keras
- - py-keras
+ # - py-keras backend=tensorflow
+ - py-keras backend=jax
+ # - py-keras backend=torch
- py-keras-applications
- py-keras-preprocessing
- py-keras2onnx