Skip to content
This repository was archived by the owner on Aug 15, 2025. It is now read-only.

Commit 88590cd

Browse files
committed
1 parent d8ece6e commit 88590cd

File tree

2 files changed

+524
-280
lines changed

2 files changed

+524
-280
lines changed

aarch64_linux/aarch64_wheel_ci_build.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -3,17 +3,19 @@
33

44
import os
55
import shutil
6-
from subprocess import check_output, check_call
7-
from pygit2 import Repository
6+
from subprocess import check_call, check_output
87
from typing import List
98

9+
from pygit2 import Repository
10+
1011

1112
def list_dir(path: str) -> List[str]:
1213
"""'
1314
Helper for getting paths for Python
1415
"""
1516
return check_output(["ls", "-1", path]).decode().split("\n")
1617

18+
1719
def build_ArmComputeLibrary() -> None:
1820
"""
1921
Using ArmComputeLibrary for aarch64 PyTorch
@@ -195,9 +197,7 @@ def parse_arguments():
195197
)
196198
if enable_cuda:
197199
desired_cuda = os.getenv("DESIRED_CUDA")
198-
build_vars += (
199-
f"BUILD_TEST=0 PYTORCH_BUILD_VERSION={version}.dev{build_date}+{desired_cuda} PYTORCH_BUILD_NUMBER=1 "
200-
)
200+
build_vars += f"BUILD_TEST=0 PYTORCH_BUILD_VERSION={version}.dev{build_date}+{desired_cuda} PYTORCH_BUILD_NUMBER=1 "
201201
else:
202202
build_vars += f"BUILD_TEST=0 PYTORCH_BUILD_VERSION={version}.dev{build_date} PYTORCH_BUILD_NUMBER=1 "
203203
elif branch.startswith(("v1.", "v2.")):

0 commit comments

Comments
 (0)