diff options
| author | Andrew W Elble <aweits@rit.edu> | 2021-04-30 19:55:44 -0400 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-04-30 18:55:44 -0500 |
| commit | bdfdba7d73c3a2ceec600e846862a5e54b98c660 (patch) | |
| tree | 2e0f37cce85705c6250ca1ef5a289fac3fd2ecdf | |
| parent | a618594c28aed14bbed9fc22a31e72221f1cc85c (diff) | |
| download | spack-bdfdba7d73c3a2ceec600e846862a5e54b98c660.tar.gz spack-bdfdba7d73c3a2ceec600e846862a5e54b98c660.tar.bz2 spack-bdfdba7d73c3a2ceec600e846862a5e54b98c660.tar.xz spack-bdfdba7d73c3a2ceec600e846862a5e54b98c660.zip | |
py-horovod: fix compilation with mxnet enabled (#23369)
| -rw-r--r-- | var/spack/repos/builtin/packages/py-horovod/package.py | 2 |
1 files changed, 2 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 0da00e92b3..5fadc26f44 100644 --- a/var/spack/repos/builtin/packages/py-horovod/package.py +++ b/var/spack/repos/builtin/packages/py-horovod/package.py @@ -171,6 +171,8 @@ class PyHorovod(PythonPackage, CudaPackage): env.set('HOROVOD_WITHOUT_PYTORCH', 1) if 'frameworks=mxnet' in self.spec: env.set('HOROVOD_WITH_MXNET', 1) + env.set('MXNET_INCLUDE_PATH', self.spec['mxnet'].prefix.include) + env.set('MXNET_LIBRARY_PATH', join_path(self.spec['mxnet'].libs[0])) else: env.set('HOROVOD_WITHOUT_MXNET', 1) |