Skip to content

Commit

Permalink
Black autoformat packaging/build_conda.py
Browse files Browse the repository at this point in the history
ghstack-source-id: 8713fa57c024d02e8c3fa7e5390b906eeb36295b
Pull Request resolved: https://github.com/fairinternal/xformers/pull/421

__original_commit__ = fairinternal/xformers@30bb37cc95931266cc64826bf0f732a0c7a7be27
  • Loading branch information
danthe3rd authored and xFormers Bot committed Jan 12, 2023
1 parent bf5a358 commit 8fd184e
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion packaging/build_conda.py
Original file line number Diff line number Diff line change
Expand Up @@ -98,7 +98,9 @@ def _set_env_for_build(self):

os.environ["TORCH_CUDA_ARCH_LIST"] = "5.0+PTX 6.0 6.1 7.0 7.5 8.0 8.6"
os.environ["BUILD_VERSION"] = self._get_build_version()
tag = subprocess.check_output(["git", "rev-parse", "--short", "HEAD"], text=True).strip()
tag = subprocess.check_output(
["git", "rev-parse", "--short", "HEAD"], text=True
).strip()
os.environ["GIT_TAG"] = tag
os.environ["PYTORCH_VERSION"] = self.pytorch_version
os.environ["CU_VERSION"] = self.cuda_version
Expand Down

0 comments on commit 8fd184e

Please sign in to comment.