From f139fff7965aec28b9542da933861a185b3025c2 Mon Sep 17 00:00:00 2001 From: Tim Moon Date: Mon, 30 Nov 2020 02:44:37 -0800 Subject: Hydrogen: disabling CPU half support also disables GPU half support (#20077) Co-authored-by: Harmen Stoppels --- var/spack/repos/builtin/packages/hydrogen/package.py | 3 +++ 1 file changed, 3 insertions(+) (limited to 'var') 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) -- cgit v1.2.3-60-g2f50