diff options
| author | Adam J. Stewart <ajstewart426@gmail.com> | 2022-01-29 20:22:19 -0600 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-01-29 20:22:19 -0600 |
| commit | 1679274ac2c959101ed79495e289e04fbc54a841 (patch) | |
| tree | 442a50c8921d2de142f2858b67c9d11dbd4b42d6 | |
| parent | c78a754e5dd99479eaeb7b9e4e668617fc405359 (diff) | |
| download | spack-1679274ac2c959101ed79495e289e04fbc54a841.tar.gz spack-1679274ac2c959101ed79495e289e04fbc54a841.tar.bz2 spack-1679274ac2c959101ed79495e289e04fbc54a841.tar.xz spack-1679274ac2c959101ed79495e289e04fbc54a841.zip | |
py-horovod: add py-tensorflow-estimator dependency (#28674)
| -rw-r--r-- | var/spack/repos/builtin/packages/py-horovod/package.py | 1 |
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') |