diff --git a/infra/tpu-pytorch-releases/artifacts.auto.tfvars b/infra/tpu-pytorch-releases/artifacts.auto.tfvars index c2617739f45..2e8db7cc7b3 100644 --- a/infra/tpu-pytorch-releases/artifacts.auto.tfvars +++ b/infra/tpu-pytorch-releases/artifacts.auto.tfvars @@ -33,6 +33,64 @@ nightly_builds = [ # Built on push to specific tag. versioned_builds = [ + # Remove libtpu from PyPI builds + { + git_tag = "v2.4.0-rc1" + package_version = "2.4.0-rc1" + pytorch_git_rev = "v2.4.0-rc1" + accelerator = "tpu" + python_version = "3.8" + bundle_libtpu = "0" + }, + { + git_tag = "v2.4.0-rc1" + package_version = "2.4.0-rc1" + pytorch_git_rev = "v2.4.0-rc1" + accelerator = "tpu" + python_version = "3.9" + bundle_libtpu = "0" + }, + { + git_tag = "v2.4.0-rc1" + package_version = "2.4.0-rc1" + pytorch_git_rev = "v2.4.0-rc1" + accelerator = "tpu" + python_version = "3.10" + bundle_libtpu = "0" + }, + { + git_tag = "v2.4.0-rc1" + package_version = "2.4.0-rc1" + pytorch_git_rev = "v2.4.0-rc1" + accelerator = "tpu" + python_version = "3.11" + bundle_libtpu = "0" + }, + # Bundle libtpu for Kaggle + { + git_tag = "v2.4.0-rc1" + package_version = "2.4.0-rc1+libtpu" + pytorch_git_rev = "v2.4.0-rc1" + accelerator = "tpu" + python_version = "3.10" + bundle_libtpu = "1" + }, + { + git_tag = "v2.4.0-rc1" + pytorch_git_rev = "v2.4.0-rc1" + package_version = "2.4.0-rc1" + accelerator = "cuda" + cuda_version = "12.1" + python_version = "3.8" + }, + { + git_tag = "v2.4.0-rc1" + pytorch_git_rev = "v2.4.0-rc1" + package_version = "2.4.0-rc1" + accelerator = "cuda" + cuda_version = "12.1" + python_version = "3.10" + }, # Remove libtpu from PyPI builds { git_tag = "v2.3.0"