Skip to content
/ carma Public

Converters between Armadillo matrices (C++) and Numpy arrays using Pybind11

License

Notifications You must be signed in to change notification settings

RUrlus/carma

Repository files navigation

carma_logo

A C++ header only library providing conversions between Numpy arrays and Armadillo matrices.

| Documentation |

Linux Build Status MacOS Build Status Windows Build Status Coverage Status Documentation Status License Release DOI

Introduction

CARMA provides fast bidirectional conversions between Numpy arrays and Armadillo matrices, vectors and cubes, much like RcppArmadillo does for R and Armadillo.

The library extends the impressive pybind11 library with support for Armadillo. For details on Pybind11 and Armadillo refer to their respective documentation 1, 2.

Installation

CARMA is a header only library that relies on two other header only libraries, Armadillo and Pybind11. It can be integrated in a CMake build using ADD_SUBDIRECTORY(<path_to_carma>) or installation which provides an interface library target carma::carma that has been linked with Python, Numpy, Pybind11 and Armadillo. See build configuration for details.

It can be installed using:

mkdir build
cd build
# optionally with -DCMAKE_INSTALL_PREFIX:PATH=<path/to/desired/location>
cmake -DCARMA_INSTALL_LIB=ON ..
cmake --build . --config Release --target install

You can than include it in a project using:

FIND_PACKAGE(carma CONFIG REQUIRED)
TARGET_LINK_LIBRARIES(<your_target> PRIVATE carma::carma)

CMake subdirectory

Alternatively you can forgo installing CARMA and directly use it as CMake subdirectory. For Pybind11 and or Armadillo we create target(s) based on user settable version, see build configuration, when they are not defined.

To link with CARMA:

ADD_SUBDIRECTORY(extern/carma)
TARGET_LINK_LIBRARIES(<your_target> PRIVATE carma::carma)

CARMA and Armadillo can then be included using:

#include <carma>
#include <armadillo>

CARMA provides a number of configurations that can be set in the carma_config.cmake file at the root of the directory or passed to CMake, see Configuration and Build configuration documentation sections for details.

Requirements

Numpy v2.* is supported by CARMA >= v0.8.0 which requires a compiler with support for C++14 and supports:

  • Python 3.8 -- 3.12
  • Numpy >= 1.14
  • Pybind11 >= v2.12.0
  • Armadillo >= 10.5.2

CARMA <= v0.7 requires a compiler with support for C++14 and supports:

  • Python 3.8 -- 3.12
  • Numpy >= 1.14 < 2.0
  • Pybind11 >= v2.6.0 < v2.12.0
  • Armadillo >= 10.5.2

CARMA makes use of Armadillo's ARMA_ALIEN_MEM_ALLOC and ARMA_ALIEN_MEM_FREE functionality introduced in version 10.5.2 to use Numpy's (de)allocator.

Considerations

In order to achieve fast conversions the default behaviour is avoid copying both from and to Numpy whenever possible and reasonable. This allows very low overhead conversions but it impacts memory safety and requires user vigilance.

If you intend to return the memory of the input array back as another array, you must make sure to either copy or steal the memory on the conversion in or copy the memory out. If you don't the memory will be aliased by the two Numpy arrays and bad things will happen.

A second consideration is memory layout. Armadillo is optimised for column-major (Fortran order) memory whereas Numpy defaults to row-major (C order). The default behaviour is to automatically convert, read copy, C-order arrays to F-order arrays upon conversion to Armadillo. Users should note that the library will not convert back to C-order when returning.

For details see the documentation section Memory Management.

Example

On a high level CARMA provides two ways to work Numpy arrays in Armadillo: Automatic conversion saves a bit on code but provides less flexibility with regards to when to copy and when not. Manual conversion should be used when you need fine grained control.

Combining the two; we use automatic conversion on the conversion in and manual when creating the tuple for the way out.

#include <carma>
#include <armadillo>
#include <pybind11/pybind11.h>
#include <pybind11/numpy.h>
#include <pybind11/pytypes.h>

namespace py = pybind11;

py::tuple ols(arma::mat& X, arma::colvec& y) {
    // We borrow the data underlying the numpy arrays
    int n = X.n_rows, k = X.n_cols;

    arma::colvec coeffs = arma::solve(X, y);
    arma::colvec resid = y - X * coeffs;

    double sig2 = arma::as_scalar(arma::trans(resid) * resid / (n-k));
    arma::colvec std_errs = arma::sqrt(sig2 * arma::diagvec( arma::inv(arma::trans(X)*X)) );

    // We take ownership of the memory from the armadillo objects and
    // return to python as a tuple containing two Numpy arrays.
    return py::make_tuple(
        carma::col_to_arr(coeffs),
        carma::col_to_arr(std_errs)
    );
}

// adapted from https://gallery.rcpp.org/articles/fast-linear-model-with-armadillo/

Which can be called using:

y = np.linspace(1, 100, num=100) + np.random.normal(0, 0.5, 100)
X = np.hstack((np.ones(100)[:, None], np.arange(1, 101)[:, None]))
coeff, std_err = carma.ols(X, y)

The repository contains tests, examples and CMake build instructions that can be used as an reference.

About

This project was created by Ralph Urlus. Significant improvements to the project have been contributed by Pascal H.

License

CARMA is provided under a Apache 2.0 license that can be found in the LICENSE file. By using, distributing, or contributing to this project, you agree to the terms and conditions of this license.