add CUDA to wrap