Skip to content

Latest commit

 

History

History
74 lines (50 loc) · 2.43 KB

INSTALLATION.md

File metadata and controls

74 lines (50 loc) · 2.43 KB
Raw

Installation

sparse_dot_topn provides wheels for CPython 3.8 to 3.12 for:

  • Windows (64bit)
  • Linux (64bit)
  • MacOS (x86 and ARM)
pip install sparse_dot_topn

sparse_dot_topn relies on a C++ extension for the computationally intensive multiplication routine. Note that the wheels vendor/ships OpenMP with the extension to provide parallelisation out-of-the-box. If you run into issue with multiple versions of OpenMP being loaded you have two options: build from source or install a non-bundled wheel.

Non-bundled wheels

We provide the same wheels without OpenMP bundled here: https://github.com/ing-bank/sparse_dot_topn/releases You than install the wheel that corresponds to your Python and OS. For example, for CPython 3.11 and MacOS ARM:

pip install sparse_dot_topn-1.0.0-cp311-cp311-macosx_11_0_arm64.whl

Building from source

Installing from source requires a C++17 compatible compiler. If you have a compiler available it is advised to install without the wheel as this enables architecture specific optimisations.

You can install from source using:

pip install sparse_dot_topn --no-binary sparse_dot_topn

Native builds

When you're building from source we assume that the target architecture is the one you are building on, aka a native build. This generally results in faster binaries at the cost that they cannot be used on different systems. Native architecture flags are enabled and the CPU is checked for support of SSE2, SSE4, AVX and AX2.

If you are building from source for a different system, you can disable this with:

CMAKE_ARGS="-DSDTN_ENABLE_ARCH_FLAGS=OFF" pip install sparse_dot_topn --no-binary sparse_dot_topn

Multithreading

Parallelisation is supported and automatically enabled if OpenMP can be found.

You can either explicitly enable OpenMP, which will now raise an exception if it cannot be found:

CMAKE_ARGS="-DSDTN_ENABLE_OPENMP=ON" pip install sparse_dot_topn --no-binary sparse_dot_topn

or explicitly disable it:

CMAKE_ARGS="-DSDTN_DISABLE_OPENMP=ON" pip install sparse_dot_topn --no-binary sparse_dot_topn

Finding OpenMP

If OpenMP cannot be found you can pass it's root directory using the OpenMP_ROOT definition. For example, for Homebrew users on Apple silicon:

CMAKE_ARGS="-DSDTN_ENABLE_OPENMP=ON -DOpenMP_ROOT=$(brew --prefix)/opt/libomp" \
pip install sparse_dot_topn --no-binary sparse_dot_topn