From 429b71bf571aa17fda0ac791e6c70961bf557a90 Mon Sep 17 00:00:00 2001 From: Ben Corbett <32752943+corbett5@users.noreply.github.com> Date: Thu, 27 May 2021 20:46:19 -0700 Subject: Fix ZFP cuda build. (#23941) --- var/spack/repos/builtin/packages/zfp/package.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) (limited to 'var') diff --git a/var/spack/repos/builtin/packages/zfp/package.py b/var/spack/repos/builtin/packages/zfp/package.py index 4d2e2006d5..e25d9b74fe 100644 --- a/var/spack/repos/builtin/packages/zfp/package.py +++ b/var/spack/repos/builtin/packages/zfp/package.py @@ -93,13 +93,21 @@ class Zfp(CMakePackage, CudaPackage): self.define_from_variant('BUILD_CFP', 'c'), self.define_from_variant('BUILD_ZFPY', 'python'), self.define_from_variant('BUILD_ZFORP', 'fortran'), + self.define_from_variant('ZFP_WITH_CUDA', 'cuda'), self.define('ZFP_BIT_STREAM_WORD_SIZE', spec.variants['bsws'].value), self.define_from_variant('ZFP_WITH_BIT_STREAM_STRIDED', 'strided'), self.define_from_variant('ZFP_WITH_ALIGNED_ALLOC', 'aligned'), self.define_from_variant('ZFP_WITH_CACHE_TWOWAY', 'twoway'), self.define_from_variant('ZFP_WITH_CACHE_FAST_HASH', 'fasthash'), - self.define_from_variant('ZFP_WITH_CACHE_PROFILE', 'profile') + self.define_from_variant('ZFP_WITH_CACHE_PROFILE', 'profile'), ] + if '+cuda' in spec: + args.append('-DCUDA_BIN_DIR={0}'.format(spec['cuda'].prefix.bin)) + + if not spec.satisfies('cuda_arch=none'): + cuda_arch = spec.variants['cuda_arch'].value + args.append('-DCMAKE_CUDA_FLAGS=-arch sm_{0}'.format(cuda_arch[0])) + return args -- cgit v1.2.3-60-g2f50