From 7d8972c8ebd3ec109589cd1406d47eda3a03401e Mon Sep 17 00:00:00 2001
From: iefgnoix <isaacwxf23@gmail.com>
Date: Fri, 22 Mar 2024 10:16:54 -0700
Subject: [PATCH] Add cuda 12.1 nightly wheels for cp310 and cp311. (#6793)

---
 infra/tpu-pytorch-releases/artifacts.auto.tfvars | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/infra/tpu-pytorch-releases/artifacts.auto.tfvars b/infra/tpu-pytorch-releases/artifacts.auto.tfvars
index f958d2e4ac8..13745551f82 100644
--- a/infra/tpu-pytorch-releases/artifacts.auto.tfvars
+++ b/infra/tpu-pytorch-releases/artifacts.auto.tfvars
@@ -19,6 +19,16 @@ nightly_builds = [
     accelerator  = "cuda"
     cuda_version = "12.1"
   },
+  {
+    accelerator  = "cuda"
+    cuda_version = "12.1"
+    python_version = "3.10"
+  },
+  {
+    accelerator  = "cuda"
+    cuda_version = "12.1"
+    python_version = "3.11"
+  },
 ]
 
 # Built on push to specific tag.