PKGBUILD


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

# Maintainer: Carlos Aznarán <caznaranl@uni.pe>
# Contributor: Will Handley <wh260@cam.ac.uk> (aur.archlinux.org/account/wjhandley)
# Based on python-mpi4py, maintained by Bruno Pagani <archange@archlinux.org> and et.al.
_base=mpi4py
pkgname=python-${_base}-intel
pkgver=4.0.1
pkgrel=1
pkgdesc="Python bindings for the Message Passing Interface compiled with intel mpi"
arch=(x86_64)
url="https://github.com/${_base}/${_base}"
license=(BSD-2-Clause)
depends=(python intel-oneapi-basekit) # intel-oneapi-dpcpp-cpp
makedepends=(python-build python-installer python-setuptools python-wheel cython)
# checkdepends=(python-pytest python-numba) # python-cupy
provides=(python-${_base})
conflicts=(python-${_base})
options=(!emptydirs)
source=(${_base}-${pkgver}.tar.gz::${url}/archive/${pkgver}.tar.gz)
sha512sums=('644471bcca595000fa8ec968e9866504f5b060966bfd798ee5ef4e954d8b849b81488028aeb74a512730a92b6f386693db2314443bbbb272a59cbc161eec5523')

build() {
  cd ${_base}-${pkgver}
  source /opt/intel/oneapi/setvars.sh
  # /opt/intel/oneapi/compiler/2024.1.0/env/vars.sh
  # /opt/intel/oneapi/mpi/2021.12/env/vars.sh
  # https://community.intel.com/t5/Intel-oneAPI-HPC-Toolkit/MPI-linking-with-LLVM-based-compilers/m-p/1498230
  # CC=icx I_MPI_CC=icx MPICC="mpiicc -cc=icx" LDFLAGS="-shared-intel"
  python -m build --wheel --skip-dependency-check --no-isolation
}

# check() {
#   # This is required starting with OpenMPI 3.0 when trying to run more
#   # processes than the number of available cores
#   export OMPI_MCA_rmaps_base_oversubscribe=yes
#   # We don’t have CUDA by default
#   export OMPI_MCA_opal_warn_on_missing_libcuda=0

#   cd ${_base}-${pkgver}
#   python -m venv --system-site-packages test-env
#   test-env/bin/python -m installer dist/*.whl
#   test-env/bin/python -m pytest -vv \
#     --ignore=test/test_cco_buf.py \
#     --ignore=test/test_cco_nb_buf.py \
#     --ignore=test/test_cco_nb_vec.py \
#     --ignore=test/test_cco_ngh_buf.py \
#     --ignore=test/test_cco_obj_inter.py \
#     --ignore=test/test_cco_vec.py \
#     --ignore=test/test_comm_inter.py \
#     --ignore=test/test_comm_inter.py \
#     --ignore=test/test_dynproc.py \
#     --ignore=test/test_io.py \
#     --ignore=test/test_msgspec.py \
#     --ignore=test/test_pack.py \
#     --ignore=test/test_p2p_buf.py \
#     --ignore=test/test_p2p_buf_matched.py \
#     --ignore=test/test_rma.py \
#     --ignore=test/test_rma_nb.py
# }

package() {
  cd ${_base}-${pkgver}
  PYTHONPYCACHEPREFIX="${PWD}/.cache/cpython/" python -m installer --destdir="${pkgdir}" dist/*.whl
  install -Dm 644 LICENSE.rst -t "${pkgdir}/usr/share/licenses/${pkgname}"
}