Bugfix

Author: xuzhao9

.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]

.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]

.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