Skip to content

Commit 7d44f86

Browse files
authored
change FA3 package matching to not promote test wheels (#7754)
as title, match on "flash_attn_3-" instead of "flash_attn_3"
1 parent bc89543 commit 7d44f86

File tree

2 files changed

+7
-1
lines changed

2 files changed

+7
-1
lines changed

.github/workflows/release-download-pytorch-org.yml

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -85,5 +85,11 @@ jobs:
8585
# Run promotion
8686
# shellcheck disable=SC2086
8787
version="${PACKAGE^^}_VERSION"
88+
89+
# do not upload FA3 test wheels with local date suffix
90+
if [[ "${PACKAGE}" == "flash_attn_3" ]]; then
91+
export PACKAGE_INCLUDE_SUFFIX="-*"
92+
fi
93+
8894
# shellcheck disable=SC2086
8995
promote_s3 ${PACKAGE} whl "${!version}"

release/promote/common_utils.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ aws_promote() {
5252
--recursive \
5353
--metadata-directive COPY \
5454
--exclude '*' \
55-
--include "*${package_name}-${pytorch_version}*" \
55+
--include "*${package_name}-${pytorch_version}${PACKAGE_INCLUDE_SUFFIX:-*}" \
5656
"${PYTORCH_S3_FROM/\/$//}" \
5757
"${PYTORCH_S3_TO/\/$//}"
5858
)

0 commit comments

Comments
 (0)