add upload var to control if docs are uploaded to s3/main

Author: clee2000

.github/workflows/_build-tutorials-base.yml

@@ -7,7 +7,12 @@ on:
         description: "Use nightly builds inside build.sh"
         required: false
         type: number
-        default: false
+        default: 0
+      UPLOAD:
+        description: "Upload built docs to PR preview and main site"
+        required: false
+        type: number
+        default: 0
 
 concurrency:
   group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.sha }}-${{ github.event_name == 'workflow_dispatch' }}-${{ github.event_name == 'schedule' }}
@@ -77,6 +82,7 @@ jobs:
           JOB_TYPE: worker
           COMMIT_SOURCE: ${{ github.ref }}
           USE_NIGHTLY: ${{ inputs.USE_NIGHTLY }}
+          UPLOAD: ${{ inputs.UPLOAD }}
         run: |
           set -ex
 
@@ -90,6 +96,7 @@ jobs:
             -e JOB_TYPE \
             -e COMMIT_SOURCE \
             -e USE_NIGHTLY \
+            -e UPLOAD \
             --env-file="/tmp/github_env_${GITHUB_RUN_ID}" \
             --tty \
             --detach \
@@ -110,6 +117,7 @@ jobs:
     name: pytorch_tutorial_build_manager
     needs: worker
     runs-on: [self-hosted, linux.2xlarge]
+    if: ${{ inputs.UPLOAD == 1 }}
     environment: ${{ github.ref == 'refs/heads/main' && 'pytorchbot-env' || '' }}
     steps:
       - name: Setup SSH (Click me for login details)

.github/workflows/build-tutorials-nightly.yml

@@ -16,3 +16,4 @@ jobs:
     secrets: inherit
     with:
       USE_NIGHTLY: 1
+      UPLOAD: 0

.github/workflows/build-tutorials.yml

@@ -16,3 +16,4 @@ jobs:
     secrets: inherit
     with:
       USE_NIGHTLY: 0
+      UPLOAD: 1

.jenkins/build.sh

@@ -20,7 +20,7 @@ sudo apt-get install -y pandoc
 # export PATH=/opt/conda/bin:$PATH
 
 # Install PyTorch Nightly for test.
-if [ "$USE_NIGHTLY" -eq 1 ]; then
+if [ "${USE_NIGHTLY:-0}" -eq 1 ]; then
   sudo pip uninstall -y torch
   pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu130
   pip show torch
@@ -120,8 +120,10 @@ if [[ "${JOB_TYPE}" == "worker" ]]; then
   python .jenkins/validate_tutorials_built.py
 
   # Step 6: Copy generated files to S3, tag with commit ID
-  7z a worker_${WORKER_ID}.7z docs
-  awsv2 s3 cp worker_${WORKER_ID}.7z s3://${BUCKET_NAME}/${COMMIT_ID}/worker_${WORKER_ID}.7z
+  if [ "${UPLOAD:-0}" -eq 1 ]; then
+    7z a worker_${WORKER_ID}.7z docs
+    awsv2 s3 cp worker_${WORKER_ID}.7z s3://${BUCKET_NAME}/${COMMIT_ID}/worker_${WORKER_ID}.7z
+  fi
 elif [[ "${JOB_TYPE}" == "manager" ]]; then
   # Step 1: Generate no-plot HTML pages for all tutorials
   pip3 install -e git+https://github.com/pytorch/pytorch_sphinx_theme.git#egg=pytorch_sphinx_theme