1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
# Copyright 2013-2020 Lawrence Livermore National Security, LLC and other
# Spack Project Developers. See the top-level COPYRIGHT file for details.
#
# SPDX-License-Identifier: (Apache-2.0 OR MIT)
from spack import *
class PyCharm4py(PythonPackage):
"""Charm4py (Charm++ for Python) is a distributed computing and parallel
programming framework for Python, for the productive development of fast,
parallel and scalable applications. It is built on top of Charm++, a C++
adaptive runtime system that has seen extensive use in the scientific and
high-performance computing (HPC) communities across many disciplines, and
has been used to develop applications that run on a wide range of devices:
from small multi-core devices up to the largest supercomputers."""
homepage = "https://charmpy.readthedocs.io"
url = "https://pypi.io/packages/source/c/charm4py/charm4py-1.0.tar.gz"
# Add a list of GitHub accounts to
# notify when the package is updated.
maintainers = ['payerle']
# Get errors passing --mpi to build* phases of setup.py
phases = ['install']
version('1.0', sha256='8ddb9f021b7379fde94b28c31f4ab6a60ced2c2a207a2d75ce57cb91b6be92bc')
variant('mpi', default=True,
description='build Charm++ library with the MPI instead of TCP'
' communication layer')
# Builds its own charm++, so no charmpp dependency
depends_on('python@2.7:2.8,3.4:', type=('build', 'run'))
depends_on('py-setuptools', type='build')
depends_on('py-cython', type='build')
depends_on('py-cffi@1.7:', type='build')
depends_on('py-numpy@1.10.0:', type=('build', 'run'))
depends_on('py-greenlet', type=('build', 'run'))
depends_on('cuda')
depends_on('mpi', when='+mpi')
# setup.py builds its own charm++, but libcharm.so
# ends up with a cuda dependency causing unresolved symbol errors
# when setup.py tries to load it to get version. We need to explicitly
# link libcudart when building the charm++ library.
# To do this, the following patch:
# 1) hacks setup.py to apply a patch to the charm++ Makefile
# causing the Makefile to include libcudart when building libcharm.so
# 2) inserts the patchfile needed to do so.
# This is convoluted, but best way I see since setup.py untars the
# charm++ sources and we need to patch a file that is in the tarball.
#
# The patch to the Makefile adds SPACK_CHARM4PY_EXTRALIBS to the link
# arguments. This needs to be set in the environment to be effective.
patch('py-charm4py.makefile.patch', when='@1.0')
# This sets the SPACK_CHARM4PY_EXTRALIBS env var which the
# py-charm4py.makefile.patch adds to the build/link command for
# libcharm.so.
def setup_build_environment(self, env):
env.set('SPACK_CHARM4PY_EXTRALIBS',
self.spec['cuda'].libs.ld_flags)
def install_args(self, spec, prefix):
# Have the parent class version set prefix
args = super().install_args(spec, prefix)
if '+mpi' in spec:
args.append('--mpi')
return args
|