Fixd CUDA build for 3.1