summaryrefslogtreecommitdiff
path: root/share
diff options
context:
space:
mode:
authorAdam J. Stewart <ajstewart426@gmail.com>2024-03-11 14:33:16 +0100
committerGitHub <noreply@github.com>2024-03-11 14:33:16 +0100
commit9786bd932bdc958bb9b6cfc55d53aa5904535c64 (patch)
treec2bc82048c23a1af793618bb796d5db898f4452f /share
parentc72619d4dbe1e9a36d43c9ef709fdff536dea9df (diff)
downloadspack-9786bd932bdc958bb9b6cfc55d53aa5904535c64.tar.gz
spack-9786bd932bdc958bb9b6cfc55d53aa5904535c64.tar.bz2
spack-9786bd932bdc958bb9b6cfc55d53aa5904535c64.tar.xz
spack-9786bd932bdc958bb9b6cfc55d53aa5904535c64.zip
Update TensorFlow ecosystem (#41069)
Diffstat (limited to 'share')
-rw-r--r--share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cpu/spack.yaml2
-rw-r--r--share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cuda/spack.yaml2
-rw-r--r--share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-rocm/spack.yaml2
3 files changed, 3 insertions, 3 deletions
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 ec28ec1fa3..fd22335885 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
@@ -21,7 +21,7 @@ spack:
- py-jaxlib
# Keras
- # - py-keras backend=tensorflow
+ - py-keras backend=tensorflow
- py-keras backend=jax
- py-keras backend=torch
- py-keras-applications
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 f9d74f6a4b..5053f28fd3 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
@@ -25,7 +25,7 @@ spack:
- py-jaxlib
# Keras
- # - py-keras backend=tensorflow
+ - py-keras backend=tensorflow
- py-keras backend=jax
- py-keras backend=torch
- py-keras-applications
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 c83b4757fc..4fb9e0eddf 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
@@ -29,7 +29,7 @@ spack:
# Keras
# - py-keras backend=tensorflow
- py-keras backend=jax
- # - py-keras backend=torch
+ - py-keras backend=torch
- py-keras-applications
- py-keras-preprocessing
- py-keras2onnx