Hack forcing the computation to stay on the GPU

Huge speedup but sadly double the ram.