Add upload of rc source code to download.pytorch.org

zklaus · zklaus · commit a17ed7493942 · 2025-04-21T21:56:55.000-06:00
ghstack-source-id: d43dd15 Pull Request resolved: #151882
diff --git a/.circleci/scripts/binary_upload.sh b/.circleci/scripts/binary_upload.sh
@@ -78,6 +78,10 @@ case "${PACKAGE_TYPE}" in
     s3_upload "zip" "libtorch"
     BACKUP_DIR="libtorch/${UPLOAD_CHANNEL}/${UPLOAD_SUBFOLDER}"
     ;;
+  source)
+    s3_upload "tar.gz" "src"
+    BACKUP_DIR="src/${UPLOAD_CHANNEL}/${UPLOAD_SUBFOLDER}"
+    ;;
   # wheel can either refer to wheel/manywheel
   *wheel)
     s3_upload "whl" "whl"
diff --git a/.github/workflows/create_release.yml b/.github/workflows/create_release.yml
@@ -108,6 +108,24 @@ jobs:
           if-no-files-found: warn
           path: ${{ needs.release.outputs.pt_release_name }}
 
+  publish_rc_source:  # Uploading
+    if: ${{ github.repository_owner == 'pytorch' }}
+    permissions:
+      id-token: write
+      contents: read
+    needs: release
+    with:
+      PYTORCH_ROOT: /pytorch
+      PACKAGE_TYPE: source
+      # TODO: This is a legacy variable that we eventually want to get rid of in
+      #       favor of GPU_ARCH_VERSION
+      DESIRED_CUDA: ""
+      GPU_ARCH_TYPE: ""
+      build_name: ${{ needs.release.outputs.pt_release_name }}
+    secrets:
+      github-token: ${{ secrets.GITHUB_TOKEN }}
+    uses: ./.github/workflows/_binary-upload.yml
+
 concurrency:
   group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.sha }}-${{ github.event_name }}
   cancel-in-progress: true
