From bdfdba7d73c3a2ceec600e846862a5e54b98c660 Mon Sep 17 00:00:00 2001 From: Andrew W Elble Date: Fri, 30 Apr 2021 19:55:44 -0400 Subject: py-horovod: fix compilation with mxnet enabled (#23369) --- var/spack/repos/builtin/packages/py-horovod/package.py | 2 ++ 1 file changed, 2 insertions(+) (limited to 'var') 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) -- cgit v1.2.3-70-g09d2