s3 binary upload with sha256 as metadata (#6247)

Equivalent of pytorch/pytorch#144887 for domains To be used in #6241
pytorch · Feb 3, 2025 · 7975e25 · 7975e25
1 parent 9892bfe
commit 7975e25
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/.github/workflows/_binary_upload.yml b/.github/workflows/_binary_upload.yml
@@ -117,7 +117,9 @@ jobs:
           fi
 
           for pkg in dist/*; do
-            ${AWS_CMD} "$pkg" "${PYTORCH_S3_BUCKET_PATH}" --acl public-read
+            shm_id=$(sha256sum "${pkg}" | awk '{print $1}')
+            ${AWS_CMD} "$pkg" "${PYTORCH_S3_BUCKET_PATH}" --acl public-read \
+              --metadata "checksum-sha256=${shm_id}"
           done
 
       - name: Upload package to pypi