[Example] Updated CUDA wrapping example (still need to test)