diff options
| -rw-r--r-- | var/spack/repos/builtin/packages/timemory/package.py | 47 |
1 files changed, 43 insertions, 4 deletions
diff --git a/var/spack/repos/builtin/packages/timemory/package.py b/var/spack/repos/builtin/packages/timemory/package.py index 717de208df..d11ac8a62c 100644 --- a/var/spack/repos/builtin/packages/timemory/package.py +++ b/var/spack/repos/builtin/packages/timemory/package.py @@ -16,16 +16,24 @@ class Timemory(CMakePackage): maintainers = ['jrmadsen'] version('master', branch='master', submodules=True) + version('develop', branch='develop', submodules=True) + version('3.0.0', commit='b36b1673b2c6b7ff3126d8261bef0f8f176c7beb', + submodules=True) variant('python', default=True, description='Enable Python support') variant('mpi', default=False, description='Enable MPI support') + variant('tau', default=True, description='Enable TAU support') variant('papi', default=True, description='Enable PAPI support') variant('cuda', default=True, description='Enable CUDA support') variant('cupti', default=True, description='Enable CUPTI support') + variant('upcxx', default=False, description='Enable UPC++ support') + variant('gotcha', default=True, description='Enable GOTCHA support') + variant('likwid', default=True, description='Enable LIKWID support') variant('caliper', default=True, description='Enable Caliper support') - variant('gperftools', default=True, description='Enable gperftools support') + variant('gperftools', default=True, + description='Enable gperftools support') - depends_on('cmake@3.10:', type='build') + depends_on('cmake@3.11:', type='build') extends('python', when='+python') depends_on('python@3:', when='+python', type=('build', 'run')) @@ -33,29 +41,40 @@ class Timemory(CMakePackage): depends_on('py-pillow', when='+python', type=('run')) depends_on('py-matplotlib', when='+python', type=('run')) depends_on('mpi', when='+mpi') + depends_on('tau', when='+tau') depends_on('papi', when='+papi') depends_on('cuda', when='+cuda') + depends_on('cuda', when='+cupti') + depends_on('upcxx', when='+upcxx') + depends_on('gotcha', when='+gotcha') + depends_on('likwid', when='+likwid') depends_on('caliper', when='+caliper') depends_on('gperftools', when='+gperftools') + conflicts('+cupti', when='~cuda', msg='CUPTI requires CUDA') + def cmake_args(self): spec = self.spec - # Use spack install of Caliper instead of internal build + # Use spack install of Caliper and/or GOTCHA + # instead of internal submodule build args = [ + '-DTIMEMORY_BUILD_GOTCHA=OFF', '-DTIMEMORY_BUILD_CALIPER=OFF', '-DTIMEMORY_BUILD_TOOLS=ON', + '-DTIMEMORY_BUILD_TESTING=OFF', '-DTIMEMORY_BUILD_EXTRA_OPTIMIZATIONS=ON', - '-DTIMEMORY_BUILD_GTEST=OFF', '-DCMAKE_INSTALL_RPATH_USE_LINK_PATH=ON', ] if '+python' in spec: args.append('-DPYTHON_EXECUTABLE={0}'.format( spec['python'].command.path)) + args.append('-DTIMEMORY_USE_PYTHON=ON') args.append('-DTIMEMORY_BUILD_PYTHON=ON') args.append('-DTIMEMORY_TLS_MODEL=global-dynamic') else: + args.append('-DTIMEMORY_USE_PYTHON=OFF') args.append('-DTIMEMORY_BUILD_PYTHON=OFF') if '+caliper' in spec: @@ -63,6 +82,16 @@ class Timemory(CMakePackage): else: args.append('-DTIMEMORY_USE_CALIPER=OFF') + if '+tau' in spec: + args.append('-DTIMEMORY_USE_TAU=ON') + else: + args.append('-DTIMEMORY_USE_TAU=OFF') + + if '+likwid' in spec: + args.append('-DTIMEMORY_USE_LIKWID=ON') + else: + args.append('-DTIMEMORY_USE_LIKWID=OFF') + if '+papi' in spec: args.append('-DTIMEMORY_USE_PAPI=ON') args.append('-DPAPI_ROOT_DIR={0}'.format(spec['papi'].prefix)) @@ -75,6 +104,16 @@ class Timemory(CMakePackage): else: args.append('-DTIMEMORY_USE_MPI=OFF') + if '+gotcha' in spec: + args.append('-DTIMEMORY_USE_GOTCHA=ON') + else: + args.append('-DTIMEMORY_USE_GOTCHA=OFF') + + if '+cuda' in spec: + args.append('-DTIMEMORY_USE_CUDA=ON') + else: + args.append('-DTIMEMORY_USE_CUDA=OFF') + if '+cupti' in spec: args.append('-DTIMEMORY_USE_CUPTI=ON') else: |