Better use of CUDA flag