gloo: add +cuda variant (#24390)

This commit is contained in:
Adam J. Stewart 2021-06-17 15:32:38 -05:00 committed by GitHub
parent d4b5911671
commit 1c22742eed
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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')]