zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Hydrogen: disabling CPU half support also disables GPU half support (#20077)

Browse Source 
Co-authored-by: Harmen Stoppels <harmenstoppels@gmail.com>
This commit is contained in:
Tim Moon 2020-11-30 02:44:37 -08:00 committed by GitHub
parent 499ada708e
commit f139fff796
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
var/spack/repos/builtin/packages/hydrogen/package.py
View File

@ -126,6 +126,8 @@ def libs(self):
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 @@ def cmake_args(self):
'-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)