diff options
-rw-r--r-- | var/spack/repos/builtin/packages/gloo/package.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/var/spack/repos/builtin/packages/gloo/package.py b/var/spack/repos/builtin/packages/gloo/package.py index 54f21c37c3..37d3d65555 100644 --- a/var/spack/repos/builtin/packages/gloo/package.py +++ b/var/spack/repos/builtin/packages/gloo/package.py @@ -6,7 +6,7 @@ from spack import * -class Gloo(CMakePackage): +class Gloo(CMakePackage, CudaPackage): """Gloo is a collective communications library.""" homepage = "https://github.com/facebookincubator/gloo" @@ -28,3 +28,6 @@ class Gloo(CMakePackage): depends_on('ninja', type='build') generator = 'Ninja' + + def cmake_args(self): + return [self.define_from_variant('USE_CUDA', 'cuda')] |