Use pytorch/test-infra#5994

Author: huydhn

.github/workflows/android-perf.yml

@@ -307,14 +307,15 @@ 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@main
+        uses: pytorch/test-infra/.github/actions/upload-benchmark-results@fix-get-job-id
         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@main
+        uses: pytorch/test-infra/.github/actions/upload-benchmark-results@fix-get-job-id
         with:
           benchmark-results-dir: benchmark-results/v3
           dry-run: false

.github/workflows/apple-perf.yml

@@ -381,14 +381,15 @@ 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@main
+        uses: pytorch/test-infra/.github/actions/upload-benchmark-results@fix-get-job-id
         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@main
+        uses: pytorch/test-infra/.github/actions/upload-benchmark-results@fix-get-job-id
         with:
           benchmark-results-dir: benchmark-results/v3
           dry-run: false