Fix compilation with cuda 8
2 files changed