diff options
| author | Tim Moon <moon13@llnl.gov> | 2020-11-30 02:44:37 -0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2020-11-30 11:44:37 +0100 |
| commit | f139fff7965aec28b9542da933861a185b3025c2 (patch) | |
| tree | 54ceaebd7b43f9bbb60f8e18444a9585fa7a3c7b /var | |
| parent | 499ada708e19018c905e2eabb18a9a41efce8240 (diff) | |
| download | spack-f139fff7965aec28b9542da933861a185b3025c2.tar.gz spack-f139fff7965aec28b9542da933861a185b3025c2.tar.bz2 spack-f139fff7965aec28b9542da933861a185b3025c2.tar.xz spack-f139fff7965aec28b9542da933861a185b3025c2.zip | |
Hydrogen: disabling CPU half support also disables GPU half support (#20077)
Co-authored-by: Harmen Stoppels <harmenstoppels@gmail.com>
Diffstat (limited to 'var')
| -rw-r--r-- | var/spack/repos/builtin/packages/hydrogen/package.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/hydrogen/package.py b/var/spack/repos/builtin/packages/hydrogen/package.py index 082d8316d7..da571027e9 100644 --- a/var/spack/repos/builtin/packages/hydrogen/package.py +++ b/var/spack/repos/builtin/packages/hydrogen/package.py @@ -126,6 +126,8 @@ class Hydrogen(CMakePackage, CudaPackage): def cmake_args(self): spec = self.spec + enable_gpu_fp16 = ('+cuda' in spec and '+half' in spec) + args = [ '-DCMAKE_INSTALL_MESSAGE:STRING=LAZY', '-DBUILD_SHARED_LIBS:BOOL=%s' % ('+shared' in spec), @@ -140,6 +142,7 @@ class Hydrogen(CMakePackage, CudaPackage): '-DHydrogen_ENABLE_CUDA=%s' % ('+cuda' in spec), '-DHydrogen_ENABLE_TESTING=%s' % ('+test' in spec), '-DHydrogen_ENABLE_HALF=%s' % ('+half' in spec), + '-DHydrogen_ENABLE_GPU_FP16=%s' % enable_gpu_fp16, ] # Add support for OS X to find OpenMP (LLVM installed via brew) |