Fix compilation without CUDA support