Improved testbench.
Minor fixes.
Started on gpu.