First draft of a working CUDA impl.
Cuda code clean up
Add cuda case to build.sh & run.sh