Fix CUDA compilation
2 files changed