Adding CUDA support, again :S