Update artifacts.auto.tfvars

pytorch · Oct 29, 2024 · 2846164 · 2846164
1 parent 2592a68
commit 2846164
Showing 1 changed file with 82 additions and 0 deletions.
diff --git a/infra/tpu-pytorch-releases/artifacts.auto.tfvars b/infra/tpu-pytorch-releases/artifacts.auto.tfvars
@@ -50,6 +50,88 @@ nightly_builds = [
 
 # Built on push to specific tag.
 versioned_builds = [
+  # Remove libtpu from PyPI builds
+  {
+    git_tag         = "v2.5.1"
+    package_version = "2.5.1"
+    pytorch_git_rev = "v2.5.1"
+    accelerator     = "tpu"
+    python_version  = "3.9"
+    bundle_libtpu   = "0"
+  },
+  {
+    git_tag         = "v2.5.1"
+    package_version = "2.5.1"
+    pytorch_git_rev = "v2.5.1"
+    accelerator     = "tpu"
+    python_version  = "3.10"
+    bundle_libtpu   = "0"
+  },
+  {
+    git_tag         = "v2.5.1"
+    package_version = "2.5.1"
+    pytorch_git_rev = "v2.5.1"
+    accelerator     = "tpu"
+    python_version  = "3.11"
+    bundle_libtpu   = "0"
+  },
+  # Bundle libtpu for Kaggle
+  {
+    git_tag         = "v2.5.1"
+    package_version = "2.5.1+libtpu"
+    pytorch_git_rev = "v2.5.1"
+    accelerator     = "tpu"
+    python_version  = "3.10"
+    bundle_libtpu   = "1"
+  },
+  {
+    git_tag         = "v2.5.1"
+    package_version = "2.5.1"
+    pytorch_git_rev = "v2.5.1"
+    accelerator     = "cuda"
+    cuda_version    = "12.1"
+    python_version  = "3.9"
+  },
+  {
+    git_tag         = "v2.5.1"
+    pytorch_git_rev = "v2.5.1"
+    package_version = "2.5.1"
+    accelerator     = "cuda"
+    cuda_version    = "12.1"
+    python_version  = "3.10"
+  },
+  {
+    git_tag         = "v2.5.1"
+    package_version = "2.5.1"
+    pytorch_git_rev = "v2.5.1"
+    accelerator     = "cuda"
+    cuda_version    = "12.1"
+    python_version  = "3.11"
+  },
+  {
+    git_tag         = "v2.5.1"
+    package_version = "2.5.1"
+    pytorch_git_rev = "v2.5.1"
+    accelerator     = "cuda"
+    cuda_version    = "12.4"
+    python_version  = "3.9"
+  },
+  {
+    git_tag         = "v2.5.1"
+    package_version = "2.5.1"
+    pytorch_git_rev = "v2.5.1"
+    accelerator     = "cuda"
+    cuda_version    = "12.4"
+    python_version  = "3.10"
+  },
+  {
+    git_tag         = "v2.5.1"
+    package_version = "2.5.1"
+    pytorch_git_rev = "v2.5.1"
+    accelerator     = "cuda"
+    cuda_version    = "12.4"
+    python_version  = "3.11"
+  },
   # Remove libtpu from PyPI builds
   {
     git_tag         = "v2.5.1-rc1"