Fixed CUDA compile error