diff --git a/setup.py b/setup.py index adaafaf741b3..f1367b850e02 100755 --- a/setup.py +++ b/setup.py @@ -277,7 +277,7 @@ def create_dir_symlink(src, dest): fd.write(f"git_hash='{git_hash}'\n") fd.write(f"git_branch='{git_branch}'\n") fd.write(f"installed_ops={install_ops}\n") - fd.write(f"accelerator_name={accelerator_name}\n") + fd.write(f"accelerator_name='{accelerator_name}'\n") fd.write(f"torch_info={torch_info}\n") print(f'install_requires={install_requires}')