Douglas-Rachford Splitting for Optimal Transport
To run experiments on the T7600's GPU:
-
Make sure the cost matrix
Chas been generated; we can do so by runing the Python notebook inexamples/. The previous step should generate acmatrixinexamples/data/. -
Check if the dimensions of
Cinexamples/test.cuare correct. If so, execute the following commands:
$ mkdir bin
$ make- For a single run:
$ ./bin/test- For performance profile:
$ ./bin/multiexp- 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.