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

tioga: add variants: +cuda and more (#21884)

Browse Source
This commit is contained in:
eugeneswalker 2021-02-25 12:47:01 -08:00 committed by GitHub
parent 177117745f
commit e8f51c6892
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 37 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

55
var/spack/repos/builtin/packages/tioga/package.py
View File

@ -2,24 +2,37 @@
# Spack Project Developers. See the top-level COPYRIGHT file for details. # Spack Project Developers. See the top-level COPYRIGHT file for details.
# #
# SPDX-License-Identifier: (Apache-2.0 OR MIT) # SPDX-License-Identifier: (Apache-2.0 OR MIT)
import sys
from spack import * from spack import *
class Tioga(CMakePackage): class Tioga(CMakePackage, CudaPackage):
"""Topology Independent Overset Grid Assembly (TIOGA)""" """Topology Independent Overset Grid Assembly (TIOGA)"""
homepage = "https://github.com/jsitaraman/tioga" homepage = "https://github.com/jsitaraman/tioga"
git = "https://github.com/jsitaraman/tioga.git" git = "https://github.com/jsitaraman/tioga.git"
maintainers = ['jsitaraman', 'sayerhs']
version('develop', branch='exawind')
version('master', branch='master') version('master', branch='master')
variant('shared', default=True, variant('shared', default=sys.platform != 'darwin',
description="Enable building shared libraries") description="Build shared libraries")
variant('pic', default=True, variant('pic', default=True,
description="Position independent code") description="Position independent code")
variant('nodegid', default=True,
description="Enable support for global Node IDs")
variant('timers', default=False,
description="Enable timers")
variant('stats', default=False,
description="Enable output of holecut stats")
variant('cxxstd', default='11',
values=('11', '14'), multi=False,
description="C++ standard to use")
depends_on('mpi') depends_on('mpi')
depends_on('cuda@9.0.0:', when='+cuda')
# Tioga has the fortran module file problem with parallel builds # Tioga has the fortran module file problem with parallel builds
parallel = False parallel = False
@ -27,20 +40,26 @@ class Tioga(CMakePackage):
def cmake_args(self): def cmake_args(self):
spec = self.spec spec = self.spec
options = [ args = [
'-DBUILD_SHARED_LIBS:BOOL=%s' % ( self.define_from_variant('BUILD_SHARED_LIBS', 'shared'),
'ON' if '+shared' in spec else 'OFF'), self.define_from_variant('CMAKE_POSITION_INDEPENDENT_CODE', 'pic'),
'-DCMAKE_POSITION_INDEPENDENT_CODE:BOOL=%s' % ( self.define_from_variant('CMAKE_CXX_STANDARD', 'cxxstd'),
'ON' if '+pic' in spec else 'OFF'), self.define_from_variant('TIOGA_HAS_NODEGID', 'nodegid'),
'-DCMAKE_C_COMPILER=%s' % spec['mpi'].mpicc, self.define_from_variant('TIOGA_ENABLE_TIMERS', 'timers'),
'-DCMAKE_CXX_COMPILER=%s' % spec['mpi'].mpicxx, self.define_from_variant('TIOGA_OUTPUT_STATS', 'stats'),
'-DCMAKE_Fortran_COMPILER=%s' % spec['mpi'].mpifc, self.define_from_variant('TIOGA_ENABLE_CUDA', 'cuda'),
'-DMPI_CXX_COMPILER:PATH=%s' % spec['mpi'].mpicxx,
'-DMPI_C_COMPILER:PATH=%s' % spec['mpi'].mpicc,
'-DMPI_Fortran_COMPILER:PATH=%s' % spec['mpi'].mpifc
] ]
if 'darwin' in spec.architecture: if '+cuda' in self.spec:
options.append('-DCMAKE_MACOSX_RPATH:BOOL=ON') args.append(self.define('CMAKE_CUDA_SEPARABLE_COMPILATION', True))
return options # Currently TIOGA only supports one device arch during specialization
cuda_arch = self.spec.variants['cuda_arch'].value
if cuda_arch:
arch_sorted = list(sorted(cuda_arch, reverse=True))
args.append(self.define('TIOGA_CUDA_SM', arch_sorted[0]))
if 'darwin' in spec.architecture:
args.append(self.define('CMAKE_MACOSX_RPATH', True))
return args