Make demo flake8-clean
Actually fix CUDA decl warning