Fixed cuda build.