CPU only, fix typo in CUDA stuff