diff --git a/infra/tpu-pytorch-releases/artifacts.auto.tfvars b/infra/tpu-pytorch-releases/artifacts.auto.tfvars index f4d9cc3b627..9ed3e33a58c 100644 --- 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"