diff options
| author | Ben Corbett <32752943+corbett5@users.noreply.github.com> | 2021-05-27 20:46:19 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-05-27 21:46:19 -0600 |
| commit | 429b71bf571aa17fda0ac791e6c70961bf557a90 (patch) | |
| tree | 423e28e46487659eadafed802c4ac56040990e93 | |
| parent | 850cac5ea3f041e9feab79f6848f98f960fc4a32 (diff) | |
| download | spack-429b71bf571aa17fda0ac791e6c70961bf557a90.tar.gz spack-429b71bf571aa17fda0ac791e6c70961bf557a90.tar.bz2 spack-429b71bf571aa17fda0ac791e6c70961bf557a90.tar.xz spack-429b71bf571aa17fda0ac791e6c70961bf557a90.zip | |
Fix ZFP cuda build. (#23941)
| -rw-r--r-- | var/spack/repos/builtin/packages/zfp/package.py | 10 |
1 files changed, 9 insertions, 1 deletions
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 |