summaryrefslogtreecommitdiff
path: root/var
diff options
context:
space:
mode:
authorAdam J. Stewart <ajstewart426@gmail.com>2022-01-29 20:22:19 -0600
committerGitHub <noreply@github.com>2022-01-29 20:22:19 -0600
commit1679274ac2c959101ed79495e289e04fbc54a841 (patch)
tree442a50c8921d2de142f2858b67c9d11dbd4b42d6 /var
parentc78a754e5dd99479eaeb7b9e4e668617fc405359 (diff)
downloadspack-1679274ac2c959101ed79495e289e04fbc54a841.tar.gz
spack-1679274ac2c959101ed79495e289e04fbc54a841.tar.bz2
spack-1679274ac2c959101ed79495e289e04fbc54a841.tar.xz
spack-1679274ac2c959101ed79495e289e04fbc54a841.zip
py-horovod: add py-tensorflow-estimator dependency (#28674)
Diffstat (limited to 'var')
-rw-r--r--var/spack/repos/builtin/packages/py-horovod/package.py1
1 files changed, 1 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/py-horovod/package.py b/var/spack/repos/builtin/packages/py-horovod/package.py
index 6bd68530e5..eed7ca386d 100644
--- a/var/spack/repos/builtin/packages/py-horovod/package.py
+++ b/var/spack/repos/builtin/packages/py-horovod/package.py
@@ -63,6 +63,7 @@ class PyHorovod(PythonPackage, CudaPackage):
# Framework dependencies
depends_on('py-tensorflow@1.1.0:', type=('build', 'link', 'run'), when='frameworks=tensorflow')
depends_on('py-tensorflow@1.15:', type=('build', 'link', 'run'), when='frameworks=tensorflow @0.20:')
+ depends_on('py-tensorflow-estimator', type=('build', 'run'), when='frameworks=tensorflow')
depends_on('py-torch@0.4.0:', type=('build', 'link', 'run'), when='frameworks=pytorch')
depends_on('py-torch@1.2:', type=('build', 'link', 'run'), when='frameworks=pytorch @0.20:')
depends_on('py-torchvision', type=('build', 'run'), when='frameworks=pytorch @:0.19.1')