8000 CUDA13 aarch64 binary seems to be missing Triton dependency · pytorch/pytorch@e91f3c6 · GitHub
[go: up one dir, main page]
Skip to content

Commit e91f3c6

Browse files
nWEIdianWEIdia
committed
CUDA13 aarch64 binary seems to be missing Triton dependency
1 parent 3e45949 commit e91f3c6

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

‎.circleci/scripts/binary_populate_env.sh‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -75,8 +75,8 @@ TRITON_VERSION=$(cat $PYTORCH_ROOT/.ci/docker/triton_version.txt)
7575
# Here PYTORCH_EXTRA_INSTALL_REQUIREMENTS is already set for the all the wheel builds hence append TRITON_CONSTRAINT
7676
TRITON_CONSTRAINT="platform_system == 'Linux' and platform_machine == 'x86_64'"
7777

78-
# CUDA 12.9 builds have triton for Linux and Linux aarch64 binaries.
79-
if [[ "$DESIRED_CUDA" == "cu129" ]]; then
78+
# CUDA 12.9/13.0 builds have triton for Linux and Linux aarch64 binaries.
79+
if [[ "$DESIRED_CUDA" == "cu129" ]] || [[ "$DESIRED_CUDA" == "cu130" ]]; then
8080
TRITON_CONSTRAINT="platform_system == 'Linux'"
8181
fi
8282

0 commit comments

Comments
 (0)
0