Skip to content

lukeolson/sparse-ops

Repository files navigation

This is an example of wrapping sparse kernels from scipy.sparse with bindthem: https://github.com/lukeolson/bindthem

Requirements

This script uses

  • OpenMP
  • omp_thread_count
  • scipy
  • tqdm

How to use

  1. Modify setup.py with your (openmp supported) compiler

  2. Build the sparse library

python setup.py build_ext --inplace
  1. Run the benchmark
python benchmark --scipy --openmp

See run.sh for a full run.

Notes

  • You need OpenMP support in your compiler. On a mac, brew install llvm libomp will work
  • Put the path for your compiler in setup.py

Example Output

THis uses analyze-data.ipynb