hlrs-software-stack/spack
7
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Timemory: Fix python dependencies + NCCL (#18342)

Browse source 
* Fix python dependencies + NCCL

* Removed trailing whitespace
This commit is contained in:
Jonathan R. Madsen 2020-08-28 12:40:34 -07:00 committed by GitHub
parent 6bb2dd40b6
commit 0bf696a29d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 14 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
var/spack/repos/builtin/packages/timemory/package.py
View file

@ -25,7 +25,10 @@ class Timemory(CMakePackage):
variant('shared', default=True, description='Build shared libraries') variant('shared', default=True, description='Build shared libraries')
variant('static', default=False, description='Build static libraries') variant('static', default=False, description='Build static libraries')
variant('python', default=False, description='Enable Python support') variant('python', default=False, description='Enable Python support')
variant('mpi', default=False, description='Enable MPI support') variant('mpi', default=False,
description='Enable support for MPI aggregation')
variant('nccl', default=False,
description='Enable support for wrapping NCCL functions')
variant('tau', default=False, description='Enable TAU support') variant('tau', default=False, description='Enable TAU support')
variant('papi', default=False, description='Enable PAPI support') variant('papi', default=False, description='Enable PAPI support')
variant('cuda', default=False, description='Enable CUDA support') variant('cuda', default=False, description='Enable CUDA support')
@ -92,7 +95,11 @@ class Timemory(CMakePackage):
depends_on('py-numpy', when='+python', type=('run')) depends_on('py-numpy', when='+python', type=('run'))
depends_on('py-pillow', when='+python', type=('run')) depends_on('py-pillow', when='+python', type=('run'))
depends_on('py-matplotlib', when='+python', type=('run')) depends_on('py-matplotlib', when='+python', type=('run'))
depends_on('py-mpi4py', when='+python+mpi', type=('run'))
depends_on('py-cython', when='+python', type=('build'))
depends_on('py-ipython', when='+python', type=('run'))
depends_on('mpi', when='+mpi') depends_on('mpi', when='+mpi')
depends_on('nccl', when='+nccl')
depends_on('tau', when='+tau') depends_on('tau', when='+tau')
depends_on('papi', when='+papi') depends_on('papi', when='+papi')
depends_on('cuda', when='+cuda') depends_on('cuda', when='+cuda')
@ -123,6 +130,8 @@ class Timemory(CMakePackage):
msg='+python require tls_model=global-dynamic') msg='+python require tls_model=global-dynamic')
conflicts('tls_model=local-exec', when='+python', conflicts('tls_model=local-exec', when='+python',
msg='+python require tls_model=global-dynamic') msg='+python require tls_model=global-dynamic')
conflicts('+nccl', when='~gotcha',
msg='+nccl requires +gotcha')
conflicts('+mpip_library', when='~mpi', msg='+mpip_library requires +mpi') conflicts('+mpip_library', when='~mpi', msg='+mpip_library requires +mpi')
conflicts('+mpip_library', when='~gotcha', conflicts('+mpip_library', when='~gotcha',
msg='+mpip_library requires +gotcha') msg='+mpip_library requires +gotcha')
@ -161,6 +170,10 @@ def cmake_args(self):
args.append('-DPYTHON_EXECUTABLE={0}'.format( args.append('-DPYTHON_EXECUTABLE={0}'.format(
spec['python'].command.path)) spec['python'].command.path))
if '+nccl' in spec:
args.append('-DTIMEMORY_USE_NCCL=ON')
args.append('-DTIMEMORY_BUILD_NCCLP_LIBRARY=ON')
if '+mpi' in spec: if '+mpi' in spec:
args.append('-DTIMEMORY_USE_MPI_LINK_FLAGS=OFF') args.append('-DTIMEMORY_USE_MPI_LINK_FLAGS=OFF')
args.append('-DMPI_C_COMPILER={0}'.format(spec['mpi'].mpicc)) args.append('-DMPI_C_COMPILER={0}'.format(spec['mpi'].mpicc))