/drot

Douglas-Rachford Splitting for Optimal Transport

Primary LanguageJupyter NotebookMIT LicenseMIT

DROT

Douglas-Rachford Splitting for Optimal Transport

To run experiments on the T7600's GPU:

  1. Make sure the cost matrix C has been generated; we can do so by runing the Python notebook in examples/. The previous step should generate a cmatrix in examples/data/.

  2. Check if the dimensions of C in examples/test.cu are correct. If so, execute the following commands:

$ mkdir bin
$ make
  • For a single run:
  $ ./bin/test
  • For performance profile:
$ ./bin/multiexp
  1. Play with the notebook

There is also the pydrot folder that contains the CPU code for sanity check. It has not been carefully optimized for performance.