Test code for CUDA