From 7bedfe6eb5e6523a34696ddd71bf3e84a3bc836e Mon Sep 17 00:00:00 2001 From: Bhavya Bahl Date: Tue, 11 Jun 2024 18:33:44 +0000 Subject: [PATCH] Add build trigger for 2.4.0-rc1 release (#7236) --- infra/ansible/config/cuda_deps.yaml | 2 + .../artifacts.auto.tfvars | 82 +++++++++++++++++++ 2 files changed, 84 insertions(+) diff --git a/infra/ansible/config/cuda_deps.yaml b/infra/ansible/config/cuda_deps.yaml index 0fcc6a0b7f5..2a6f0d33888 100644 --- a/infra/ansible/config/cuda_deps.yaml +++ b/infra/ansible/config/cuda_deps.yaml @@ -3,6 +3,7 @@ cuda_deps: # List all libcudnn8 versions with `apt list -a libcudnn8` libcudnn: + "12.4": libcudnn-cuda-12=9.1.1.17-1 "12.3": libcudnn9-cuda-12=9.0.0.312-1 "12.1": libcudnn8=8.9.2.26-1+cuda12.1 "12.0": libcudnn8=8.8.0.121-1+cuda12.0 @@ -10,6 +11,7 @@ cuda_deps: "11.7": libcudnn8=8.5.0.96-1+cuda11.7 "11.2": libcudnn8=8.1.1.33-1+cuda11.2 libcudnn-dev: + "12.4": libcudnn-dev-cuda-12=9.1.1.17-1 "12.3": libcudnn9-dev-cuda-12=9.0.0.312-1 "12.1": libcudnn8-dev=8.9.2.26-1+cuda12.1 "12.0": libcudnn8-dev=8.8.0.121-1+cuda12.0 diff --git a/infra/tpu-pytorch-releases/artifacts.auto.tfvars b/infra/tpu-pytorch-releases/artifacts.auto.tfvars index c2617739f45..833393e90aa 100644 --- a/infra/tpu-pytorch-releases/artifacts.auto.tfvars +++ b/infra/tpu-pytorch-releases/artifacts.auto.tfvars @@ -33,6 +33,88 @@ 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.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" + package_version = "2.4.0-rc1" + pytorch_git_rev = "v2.4.0-rc1" + accelerator = "cuda" + cuda_version = "12.1" + python_version = "3.9" + }, + { + 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" + }, + { + git_tag = "v2.4.0-rc1" + package_version = "2.4.0-rc1" + pytorch_git_rev = "v2.4.0-rc1" + accelerator = "cuda" + cuda_version = "12.1" + python_version = "3.11" + }, + { + git_tag = "v2.4.0-rc1" + package_version = "2.4.0-rc1" + pytorch_git_rev = "v2.4.0-rc1" + accelerator = "cuda" + cuda_version = "12.4" + python_version = "3.9" + }, + { + git_tag = "v2.4.0-rc1" + package_version = "2.4.0-rc1" + pytorch_git_rev = "v2.4.0-rc1" + accelerator = "cuda" + cuda_version = "12.4" + python_version = "3.10" + }, + { + git_tag = "v2.4.0-rc1" + package_version = "2.4.0-rc1" + pytorch_git_rev = "v2.4.0-rc1" + accelerator = "cuda" + cuda_version = "12.4" + python_version = "3.11" + }, # Remove libtpu from PyPI builds { git_tag = "v2.3.0"