Fixed compilation issue with no cuda.