rwolst / sparse-cholesky-project

Ultimately add sparse Cholesky op to Tensorflow

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Sparse Cholesky Project

A project that attempts to do the following:

  • Wrap the Eigen sparse (and conjugate gradient) Cholesky solver for Python.
  • Test speed against the scikits-spase wrapper for CHOLMOD on multiple benchmarks.
  • Create a Tensorflow op for sparse Cholesky solver.
  • Create the Tensorflow forward and backward differentation equations.

Installation

Eigen

To install Eigen, simply download the source from here and copy the header files into a location that is in the PATH e.g.

cd ~/Downloads
tar -xvf ~/eigen-eigen-<build-info>
cd eigen-eigen-<build-info>
sudo cp -r Eigen /usr/local/include

Usage

For example

python3 test.py 10000 x x cholesky_dense numpy

About

Ultimately add sparse Cholesky op to Tensorflow


Languages

Language:Python 65.4%Language:C++ 31.7%Language:C 2.9%