fixed compilation with CUDA