From 8269b80a4ea7bb00811f8659695e7f5b7789dc78 Mon Sep 17 00:00:00 2001
From: DanilBaibak <baibak@meta.com>
Date: Mon, 16 Oct 2023 10:19:23 +0200
Subject: [PATCH] Added a cleanup step to all macOS workflows

---
 .github/workflows/build_conda_macos.yml  | 4 ++++
 .github/workflows/build_wheels_macos.yml | 4 ++++
 .github/workflows/macos_job.yml          | 4 ++++
 3 files changed, 12 insertions(+)

diff --git a/.github/workflows/build_conda_macos.yml b/.github/workflows/build_conda_macos.yml
index f6bdfa3fab..e2e9f3de1d 100644
--- a/.github/workflows/build_conda_macos.yml
+++ b/.github/workflows/build_conda_macos.yml
@@ -232,6 +232,10 @@ jobs:
             --label main \
             --no-progress \
             --force
+      - name: Clean up disk space
+        if: always()
+        continue-on-error: true
+        uses: ./test-infra/.github/actions/check-disk-space@main
 
 concurrency:
   group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.sha }}-${{ inputs.repository }}-${{ github.event_name == 'workflow_dispatch' }}
diff --git a/.github/workflows/build_wheels_macos.yml b/.github/workflows/build_wheels_macos.yml
index db62377562..1e10531f3c 100644
--- a/.github/workflows/build_wheels_macos.yml
+++ b/.github/workflows/build_wheels_macos.yml
@@ -214,6 +214,10 @@ jobs:
             # PYTORCH_S3_BUCKET_PATH derived from pkg-helpers
             ${CONDA_RUN} aws s3 cp "$pkg" "${PYTORCH_S3_BUCKET_PATH}" --acl public-read
           done
+      - name: Clean up disk space
+        if: always()
+        continue-on-error: true
+        uses: ./test-infra/.github/actions/check-disk-space@main
 
 concurrency:
   group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.sha }}-${{ inputs.repository }}-${{ github.event_name == 'workflow_dispatch' }}
diff --git a/.github/workflows/macos_job.yml b/.github/workflows/macos_job.yml
index c61ffc9d80..74442c0615 100644
--- a/.github/workflows/macos_job.yml
+++ b/.github/workflows/macos_job.yml
@@ -178,3 +178,7 @@ jobs:
           name: ${{ inputs.upload-artifact }}
           path: ${{ runner.temp }}/artifacts/
           if-no-files-found: ${{ steps.check-artifacts.outputs.if-no-files-found }}
+      - name: Clean up disk space
+        if: always()
+        continue-on-error: true
+        uses: ./test-infra/.github/actions/check-disk-space@main