Add Python 3.9 build for 2.1 release (pytorch#5744)

chunnienc · Dec 14, 2023 · f911118 · f911118
1 parent 379c6f2
commit f911118
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/infra/tpu-pytorch-releases/artifacts.auto.tfvars b/infra/tpu-pytorch-releases/artifacts.auto.tfvars
@@ -49,13 +49,22 @@ xrt_versioned_builds = [
 
 # Built on push to specific tag.
 versioned_builds = [
+  # Remove libtpu from PyPI builds
   {
     git_tag         = "v2.1.0"
     pytorch_git_rev = "v2.1.0"
     package_version = "2.1.0"
     accelerator     = "tpu"
     bundle_libtpu   = "0"
   },
+  {
+    git_tag         = "v2.1.0"
+    pytorch_git_rev = "v2.1.0"
+    package_version = "2.1.0"
+    accelerator     = "tpu"
+    python_version  = "3.9"
+    bundle_libtpu   = "0"
+  },
   {
     git_tag         = "v2.1.0"
     pytorch_git_rev = "v2.1.0"
@@ -64,6 +73,7 @@ versioned_builds = [
     python_version  = "3.10"
     bundle_libtpu   = "0"
   },
+  # Bundle libtpu for Kaggle
   {
     git_tag         = "v2.1.0"
     pytorch_git_rev = "v2.1.0"