fixed build with CUDA after latest merge