Fix compiling for non-CUDA machines caused by bb64d063191aa830028d01b7692fc94d21510439