Ready to land

Author: huydhn

.github/workflows/android-perf.yml

@@ -307,15 +307,14 @@ jobs:
 
       # TODO (huydhn): Remove v2 schema once the benchmark dashboard finishes the migration
       - name: Upload the benchmark results (v2)
-        uses: pytorch/test-infra/.github/actions/upload-benchmark-results@fix-get-job-id
+        uses: pytorch/test-infra/.github/actions/upload-benchmark-results@main
         with:
           benchmark-results-dir: benchmark-results/v2
           dry-run: false
           schema-version: v2
 
-      # TODO: Switch back to main after https://github.com/pytorch/test-infra/pull/5994 lands
       - name: Upload the benchmark results (v3)
-        uses: pytorch/test-infra/.github/actions/upload-benchmark-results@fix-get-job-id
+        uses: pytorch/test-infra/.github/actions/upload-benchmark-results@main
         with:
           benchmark-results-dir: benchmark-results/v3
           dry-run: false

.github/workflows/apple-perf.yml

@@ -381,15 +381,14 @@ jobs:
 
       # TODO (huydhn): Remove v2 schema once the benchmark dashboard finishes the migration
       - name: Upload the benchmark results (v2)
-        uses: pytorch/test-infra/.github/actions/upload-benchmark-results@fix-get-job-id
+        uses: pytorch/test-infra/.github/actions/upload-benchmark-results@main
         with:
           benchmark-results-dir: benchmark-results/v2
           dry-run: false
           schema-version: v2
 
-      # TODO: Switch back to main after https://github.com/pytorch/test-infra/pull/5994 lands
       - name: Upload the benchmark results (v3)
-        uses: pytorch/test-infra/.github/actions/upload-benchmark-results@fix-get-job-id
+        uses: pytorch/test-infra/.github/actions/upload-benchmark-results@main
         with:
           benchmark-results-dir: benchmark-results/v3
           dry-run: false