diff options
Diffstat (limited to 'var')
| -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) |