diff --git a/.gitignore b/.gitignore
index 85e6672ea629461c92dc03e64ef75440d8b6a794..5d08542451279ce32183e15482f7af4586283862 100644
--- a/.gitignore
+++ b/.gitignore
@@ -68,3 +68,7 @@ commit.msg
# Mac default files
.DS_Store
+
+# by product of generating a source distribution for pip
+*.egg-info
+dist
diff --git a/MANIFEST.in b/MANIFEST.in
new file mode 100644
index 0000000000000000000000000000000000000000..58cd8c0b66e30a6f443d5924196f4f6205d3d443
--- /dev/null
+++ b/MANIFEST.in
@@ -0,0 +1,17 @@
+include VERSION
+include CMakeLists.txt
+
+graft arbor
+graft arborenv
+graft cmake
+graft doc
+graft example
+graft ext
+graft mechanisms
+graft modcc
+graft python
+graft sup
+graft test
+
+prune ext/sphinx_rtd_theme
+prune ext/google-benchmark
diff --git a/setup.py b/setup.py
index 30214847d1cfc418a74fd408dc35902a8891adaa..131304a9e051d347ee5ff861863efb61701f3f69 100644
--- a/setup.py
+++ b/setup.py
@@ -12,6 +12,10 @@ here = os.path.abspath(os.path.dirname(__file__))
with open(os.path.join(here, 'VERSION')) as version_file:
version_ = version_file.read().strip()
+# Get the contents of the readme
+with open(os.path.join(here, 'python/readme.md'), encoding='utf-8') as f:
+ long_description = f.read()
+
def check_cmake():
try:
out = subprocess.check_output(['cmake', '--version'])
@@ -42,18 +46,20 @@ class install_command(install):
install.finalize_options(self)
def run(self):
- # The options are stored in a dictionary cl_opt, with the following keys:
- # 'mpi' : build with MPI support (boolean).
- # 'gpu' : build with CUDA support (boolean).
- # 'vec' : generate SIMD vectorized kernels for CPU micro-architecture (boolean).
- # 'arch' : target CPU micro-architecture (string).
- global cl_opt
- cl_opt = {
- 'mpi' : self.mpi is not None,
- 'gpu' : self.gpu is not None,
- 'vec' : self.vec is not None,
- 'arch': "native" if self.arch is None else self.arch
- }
+ # The options are stored in global variables:
+ # cl_opt_mpi : build with MPI support (boolean).
+ # cl_opt_gpu : build with CUDA support (boolean).
+ # cl_opt_vec : generate SIMD vectorized kernels for CPU micro-architecture (boolean).
+ # cl_opt_arch : target CPU micro-architecture (string).
+ global cl_opt_mpi
+ global cl_opt_gpu
+ global cl_opt_vec
+ global cl_opt_arch
+ cl_opt_mpi = self.mpi is not None
+ cl_opt_gpu = self.gpu is not None
+ cl_opt_vec = self.vec is not None
+ cl_opt_arch = "native" if self.arch is None else self.arch
+
install.run(self)
class cmake_extension(Extension):
@@ -65,22 +71,6 @@ class cmake_build(build_ext):
if not check_cmake():
raise RuntimeError('CMake is not available. CMake 3.12 is required.')
- # cl_opt contains the command line arguments passed to install, via
- # --install-option if using pip.
- # pip skips building wheels when --install-option flags are set.
- # However, when no --install-options are passed, it runs build_ext
- # without running install_command, required to create and set cl_opt.
- # This hack works around this. I think that the upshot of this is
- # that only wheels built with default configuration will be possible.
-
- if 'cl_opt' not in globals():
- cl_opt = {
- 'mpi': False,
- 'gpu': False,
- 'vec': False,
- 'arch': 'native'
- }
-
# The path where CMake will be configured and Arbor will be built.
build_directory = os.path.abspath(self.build_temp)
# The path where the package will be copied after building.
@@ -94,24 +84,19 @@ class cmake_build(build_ext):
cmake_args = [
'-DARB_WITH_PYTHON=on',
'-DPYTHON_EXECUTABLE=' + sys.executable,
- '-DARB_WITH_MPI={}'.format('on' if cl_opt['mpi'] else 'off'),
- '-DARB_WITH_GPU={}'.format('on' if cl_opt['gpu'] else 'off'),
- '-DARB_VECTORIZE={}'.format('on' if cl_opt['vec'] else 'off'),
- '-DARB_ARCH={}'.format(cl_opt['arch']),
+ '-DARB_WITH_MPI={}'.format( 'on' if cl_opt_mpi else 'off'),
+ '-DARB_WITH_GPU={}'.format( 'on' if cl_opt_gpu else 'off'),
+ '-DARB_VECTORIZE={}'.format('on' if cl_opt_vec else 'off'),
+ '-DARB_ARCH={}'.format(cl_opt_arch),
+ '-DCMAKE_BUILD_TYPE=Release' # we compile with debug symbols in release mode.
]
- print('-'*5, 'command line options: {}'.format(cl_opt))
print('-'*5, 'cmake arguments: {}'.format(cmake_args))
- cfg = 'Debug' if self.debug else 'Release'
- build_args = ['--config', cfg]
-
- cmake_args += ['-DCMAKE_BUILD_TYPE=' + cfg]
+ build_args = ['--config', 'Release']
# Assuming Makefiles
- build_args += ['--', '-j4']
-
- self.build_args = build_args
+ build_args += ['--', '-j2']
env = os.environ.copy()
env['CXXFLAGS'] = '{}'.format(env.get('CXXFLAGS', ''))
@@ -125,7 +110,7 @@ class cmake_build(build_ext):
cwd=self.build_temp, env=env)
print('-'*20, 'Build')
- cmake_cmd = ['cmake', '--build', '.'] + self.build_args
+ cmake_cmd = ['cmake', '--build', '.'] + build_args
subprocess.check_call(cmake_cmd,
cwd=self.build_temp)
@@ -153,12 +138,11 @@ setuptools.setup(
author='The lovely Arbor devs.',
url='https://github.com/arbor-sim/arbor',
description='High performance simulation of networks of multicompartment neurons.',
- long_description='',
+ long_description=long_description,
+ long_description_content_type='text/markdown',
classifiers=[
'Development Status :: 4 - Beta', # Upgrade to "5 - Production/Stable" on release of v0.3
'Intended Audience :: Science/Research',
- 'Topic :: Scientific/Engineering :: Build Tools',
- 'License :: OSI Approved :: BSD License'
'Programming Language :: Python :: 3.6',
'Programming Language :: Python :: 3.7',
'Programming Language :: Python :: 3.8',