Hydrogen: disabling CPU half support also disables GPU half support (#20077)
Co-authored-by: Harmen Stoppels <harmenstoppels@gmail.com>
This commit is contained in:
parent
499ada708e
commit
f139fff796
@ -126,6 +126,8 @@ def libs(self):
|
|||||||
def cmake_args(self):
|
def cmake_args(self):
|
||||||
spec = self.spec
|
spec = self.spec
|
||||||
|
|
||||||
|
enable_gpu_fp16 = ('+cuda' in spec and '+half' in spec)
|
||||||
|
|
||||||
args = [
|
args = [
|
||||||
'-DCMAKE_INSTALL_MESSAGE:STRING=LAZY',
|
'-DCMAKE_INSTALL_MESSAGE:STRING=LAZY',
|
||||||
'-DBUILD_SHARED_LIBS:BOOL=%s' % ('+shared' in spec),
|
'-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_CUDA=%s' % ('+cuda' in spec),
|
||||||
'-DHydrogen_ENABLE_TESTING=%s' % ('+test' in spec),
|
'-DHydrogen_ENABLE_TESTING=%s' % ('+test' in spec),
|
||||||
'-DHydrogen_ENABLE_HALF=%s' % ('+half' 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)
|
# Add support for OS X to find OpenMP (LLVM installed via brew)
|
||||||
|
Loading…
Reference in New Issue
Block a user