From ce68c0086de6d9b5e824f80d1177457012229721 Mon Sep 17 00:00:00 2001
From: Nicolas Hug <contact@nicolas-hug.com>
Date: Fri, 10 Feb 2023 11:32:03 +0000
Subject: [PATCH] Try GPU tests on 11.6 and 11.8

---
 .github/workflows/test-linux-gpu.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/test-linux-gpu.yml b/.github/workflows/test-linux-gpu.yml
index 831de27e350..c18131c6336 100644
--- a/.github/workflows/test-linux-gpu.yml
+++ b/.github/workflows/test-linux-gpu.yml
@@ -17,7 +17,7 @@ jobs:
     strategy:
       matrix:
         python_version: ["3.8"]
-        cuda_arch_version: ["11.7"]
+        cuda_arch_version: ["11.6", "11.8"]
       fail-fast: false
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
     with: