From e06fbaf95363714dbd7edaf84f3154ac9e36416e Mon Sep 17 00:00:00 2001
From: Xu Zhao <xzhao9@meta.com>
Date: Fri, 21 Jun 2024 18:28:42 -0400
Subject: [PATCH] Bugfix

---
 .github/workflows/_linux-test-cpu.yml  |  2 +-
 .github/workflows/_linux-test-cuda.yml |  2 +-
 .github/workflows/pr-test.yml          | 17 ++++++++---------
 3 files changed, 10 insertions(+), 11 deletions(-)

diff --git a/.github/workflows/_linux-test-cpu.yml b/.github/workflows/_linux-test-cpu.yml
index 4b46a5bcc..228ae6aa3 100644
--- a/.github/workflows/_linux-test-cpu.yml
+++ b/.github/workflows/_linux-test-cpu.yml
@@ -1,7 +1,7 @@
 name: linux-test-cpu
 
 jobs:
-  linux-pr-test-cpu:
+  linux-test-cpu:
     # Don't run on forked repos
     if: github.repository_owner == 'pytorch'
     runs-on: [self-hosted, linux.24xlarge]
diff --git a/.github/workflows/_linux-test-cuda.yml b/.github/workflows/_linux-test-cuda.yml
index 1bd0f8275..b8b531436 100644
--- a/.github/workflows/_linux-test-cuda.yml
+++ b/.github/workflows/_linux-test-cuda.yml
@@ -1,7 +1,7 @@
 name: linux-test-cuda
 
 jobs:
-  linux-pr-test-cuda:
+  linux-test-cuda:
     # Don't run on forked repos
     if: github.repository_owner == 'pytorch'
     runs-on: [a100-runner]
diff --git a/.github/workflows/pr-test.yml b/.github/workflows/pr-test.yml
index cf53acd18..ea5dca371 100644
--- a/.github/workflows/pr-test.yml
+++ b/.github/workflows/pr-test.yml
@@ -6,17 +6,16 @@ on:
     branches:
       - main
 
+concurrency:
+  group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref_name }}-${{ github.ref_type == 'branch' && github.sha }}-${{ github.event_name == 'workflow_dispatch' }}-${{ github.event_name == 'schedule' }}
+  cancel-in-progress: true
+
 jobs:
   cpu-test:
-    timeout-minutes: 120 # 2 hours
-    uses: ./.github/workflow/_linux-test-cpu.yml
+    uses: ./.github/workflows/_linux-test-cpu.yml
     with:
-      timeout-minutes: 120 # 2 hours
+      timeout-minutes: 240 # 4 hours
   cuda-test:
-    uses: ./.github/workflow/_linux-test-cuda.yml
+    uses: ./.github/workflows/_linux-test-cuda.yml
     with:
-      timeout-minutes: 120 # 2 hours
-
-concurrency:
-  group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.sha }}-${{ github.event_name == 'workflow_dispatch' }}
-  cancel-in-progress: true
+      timeout-minutes: 240 # 4 hours