Update

[ghstack-poisoned]

Author: jainapurva

.github/workflows/float8_test.yml

@@ -27,12 +27,12 @@ jobs:
             runs-on: linux.g6.4xlarge.experimental.nvidia.gpu
             torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/cu124'
             gpu-arch-type: "cuda"
-            gpu-arch-version: "12.4"
+            gpu-arch-version: "12.6"
           - name: H100
             runs-on: linux.aws.h100
             torch-spec: '--pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124'
             gpu-arch-type: "cuda"
-            gpu-arch-version: "12.4"
+            gpu-arch-version: "12.6"
 
     permissions:
       id-token: write

.github/workflows/float8nocompile_test.yaml

@@ -31,7 +31,7 @@ env:
 #             runs-on: linux.aws.h100
 #             torch-spec: '--pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124'
 #             gpu-arch-type: "cuda"
-#             gpu-arch-version: "12.4"
+#             gpu-arch-version: "12.6"
 
 #     uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
 #     with:

.github/workflows/nightly_smoke_test.yml

@@ -23,7 +23,7 @@ jobs:
             runs-on: linux.g5.12xlarge.nvidia.gpu
             torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/cu124'
             gpu-arch-type: "cuda"
-            gpu-arch-version: "12.4"
+            gpu-arch-version: "12.6"
 
     permissions:
       id-token: write

.github/workflows/regression_test.yml

@@ -27,7 +27,7 @@ jobs:
             runs-on: linux.g5.12xlarge.nvidia.gpu
             torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/cu124'
             gpu-arch-type: "cuda"
-            gpu-arch-version: "12.4"
+            gpu-arch-version: "12.6"
           - name: CPU Nightly
             runs-on: linux.4xlarge
             torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/cpu'
@@ -63,17 +63,17 @@ jobs:
             runs-on: linux.g5.12xlarge.nvidia.gpu
             torch-spec: 'torch==2.3.0'
             gpu-arch-type: "cuda"
-            gpu-arch-version: "12.1"
+            gpu-arch-version: "12.6"
           - name: CUDA 2.4
             runs-on: linux.g5.12xlarge.nvidia.gpu
             torch-spec: 'torch==2.4.0'
             gpu-arch-type: "cuda"
-            gpu-arch-version: "12.1"
+            gpu-arch-version: "12.6"
           - name: CUDA 2.5.1
             runs-on: linux.g5.12xlarge.nvidia.gpu
             torch-spec: 'torch==2.5.1 --index-url https://download.pytorch.org/whl/cu121'
             gpu-arch-type: "cuda"
-            gpu-arch-version: "12.1"
+            gpu-arch-version: "12.6"
 
           - name: CPU 2.3
             runs-on: linux.4xlarge