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

lbann: allow unsupported compiler (#29417)

Browse Source
This commit is contained in:
Brian Van Essen 2022-03-09 10:47:55 -08:00 committed by GitHub
parent 033647c8b1
commit ff7230c0cf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 18 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
var/spack/repos/builtin/packages/aluminum/package.py
View File

@ -81,6 +81,10 @@ def cmake_args(self):
arch_str = ";".join(archs) arch_str = ";".join(archs)
args.append('-DCMAKE_CUDA_ARCHITECTURES=%s' % arch_str) args.append('-DCMAKE_CUDA_ARCHITECTURES=%s' % arch_str)
if (spec.satisfies('%cce') and
spec.satisfies('^cuda+allow-unsupported-compilers')):
args.append('-DCMAKE_CUDA_FLAGS=-allow-unsupported-compiler')
if spec.satisfies('@0.5:'): if spec.satisfies('@0.5:'):
args.extend([ args.extend([
'-DALUMINUM_ENABLE_HOST_TRANSFER:BOOL=%s' % ('+ht' in spec), '-DALUMINUM_ENABLE_HOST_TRANSFER:BOOL=%s' % ('+ht' in spec),

4
var/spack/repos/builtin/packages/dihydrogen/package.py
View File

@ -167,6 +167,10 @@ def cmake_args(self):
arch_str = ";".join(archs) arch_str = ";".join(archs)
args.append('-DCMAKE_CUDA_ARCHITECTURES=%s' % arch_str) args.append('-DCMAKE_CUDA_ARCHITECTURES=%s' % arch_str)
if (spec.satisfies('%cce') and
spec.satisfies('^cuda+allow-unsupported-compilers')):
args.append('-DCMAKE_CUDA_FLAGS=-allow-unsupported-compiler')
if '+cuda' in spec or '+distconv' in spec: if '+cuda' in spec or '+distconv' in spec:
args.append('-DcuDNN_DIR={0}'.format( args.append('-DcuDNN_DIR={0}'.format(
spec['cudnn'].prefix)) spec['cudnn'].prefix))

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

@ -69,6 +69,8 @@ class Hydrogen(CMakePackage, CudaPackage, ROCmPackage):
depends_on('cmake@3.21.0:', type='build', when='@1.5.2:') depends_on('cmake@3.21.0:', type='build', when='@1.5.2:')
depends_on('cmake@3.17.0:', type='build', when='@:1.5.1') depends_on('cmake@3.17.0:', type='build', when='@:1.5.1')
depends_on('cmake@3.22.0:', type='build', when='%cce')
depends_on('mpi') depends_on('mpi')
depends_on('hwloc@1.11:') depends_on('hwloc@1.11:')
depends_on('hwloc +cuda +nvml', when='+cuda') depends_on('hwloc +cuda +nvml', when='+cuda')
@ -176,6 +178,10 @@ def cmake_args(self):
arch_str = ";".join(archs) arch_str = ";".join(archs)
args.append('-DCMAKE_CUDA_ARCHITECTURES=%s' % arch_str) args.append('-DCMAKE_CUDA_ARCHITECTURES=%s' % arch_str)
if (spec.satisfies('%cce') and
spec.satisfies('^cuda+allow-unsupported-compilers')):
args.append('-DCMAKE_CUDA_FLAGS=-allow-unsupported-compiler')
if '+rocm' in spec: if '+rocm' in spec:
args.extend([ args.extend([
'-DCMAKE_CXX_FLAGS=-std=c++17', '-DCMAKE_CXX_FLAGS=-std=c++17',

4
var/spack/repos/builtin/packages/lbann/package.py
View File

@ -325,6 +325,10 @@ def cmake_args(self):
arch_str = ";".join(archs) arch_str = ";".join(archs)
args.append('-DCMAKE_CUDA_ARCHITECTURES=%s' % arch_str) args.append('-DCMAKE_CUDA_ARCHITECTURES=%s' % arch_str)
if (spec.satisfies('%cce') and
spec.satisfies('^cuda+allow-unsupported-compilers')):
args.append('-DCMAKE_CUDA_FLAGS=-allow-unsupported-compiler')
if spec.satisfies('@:0.90') or spec.satisfies('@0.95:'): if spec.satisfies('@:0.90') or spec.satisfies('@0.95:'):
args.append( args.append(
'-DHydrogen_DIR={0}/CMake/hydrogen'.format( '-DHydrogen_DIR={0}/CMake/hydrogen'.format(