CUDA fixes