CUDA backend fixes