From 4373bbd6314fb8eec3585358030bb3b5be9bcf5c Mon Sep 17 00:00:00 2001
From: Nicolas Richart <nicolas.richart@epfl.ch>
Date: Tue, 6 Aug 2024 17:24:07 +0200
Subject: [PATCH] "Correcting" py-horovod
---
stacks/pinot-noir-gcc13/packages_stack.yaml | 3 +++
stacks/pinot-noir-gcc13/spack.yaml | 2 +-
2 files changed, 4 insertions(+), 1 deletion(-)
diff --git a/stacks/pinot-noir-gcc13/packages_stack.yaml b/stacks/pinot-noir-gcc13/packages_stack.yaml
index 5f27d89..27dc28a 100644
--- a/stacks/pinot-noir-gcc13/packages_stack.yaml
+++ b/stacks/pinot-noir-gcc13/packages_stack.yaml
@@ -214,6 +214,9 @@ packages:
py-fenics-dolfinx:
require: ['@0.8.0']
+ py-horovod:
+ require: ['frameworks=pytorch,keras,tensorflow', 'controllers=mpi', 'tensor_ops=nccl']
+
py-numpy:
prefer: ['@1.26.4']
diff --git a/stacks/pinot-noir-gcc13/spack.yaml b/stacks/pinot-noir-gcc13/spack.yaml
index 5b1631e..73525a4 100644
--- a/stacks/pinot-noir-gcc13/spack.yaml
+++ b/stacks/pinot-noir-gcc13/spack.yaml
@@ -297,7 +297,7 @@ spack:
- paraview@5.12.0 +shared +python +hdf5 +opengl2
- py-fenics-dolfinx
- py-h5py ++mpi
- - py-horovod
+ # - py-horovod cannot resolve dependencies with pytorch
- py-keras
- py-tensorflow
- py-torch
--
GitLab