From 4dd6931f05c28344ac47a20eaf53f69360b41cb0 Mon Sep 17 00:00:00 2001 From: Will Cromar Date: Wed, 13 Sep 2023 22:54:37 +0000 Subject: [PATCH 1/3] Add bundled libtpu build for Python 3.10 --- infra/tpu-pytorch-releases/artifacts.auto.tfvars | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/infra/tpu-pytorch-releases/artifacts.auto.tfvars b/infra/tpu-pytorch-releases/artifacts.auto.tfvars index 2982d3f2415a..130d1b3bf2f0 100644 --- a/infra/tpu-pytorch-releases/artifacts.auto.tfvars +++ b/infra/tpu-pytorch-releases/artifacts.auto.tfvars @@ -45,6 +45,7 @@ versioned_builds = [ pytorch_git_rev = "v2.1.0-rc2" package_version = "2.1.0rc2" accelerator = "tpu" + bundle_libtpu = "0" }, { git_tag = "v2.1.0" @@ -54,6 +55,13 @@ versioned_builds = [ python_version = "3.10" bundle_libtpu = "0" }, + { + git_tag = "v2.1.0" + pytorch_git_rev = "v2.1.0-rc2" + package_version = "2.1.0rc2+libtpu" + accelerator = "tpu" + python_version = "3.10" + }, { git_tag = "v2.0.0" package_version = "2.0" From 8b0db4dfb7a19217c2eb690e8637636abe8093cd Mon Sep 17 00:00:00 2001 From: Will Cromar Date: Wed, 13 Sep 2023 22:55:24 +0000 Subject: [PATCH 2/3] make bundle explicit --- infra/tpu-pytorch-releases/artifacts.auto.tfvars | 1 + 1 file changed, 1 insertion(+) diff --git a/infra/tpu-pytorch-releases/artifacts.auto.tfvars b/infra/tpu-pytorch-releases/artifacts.auto.tfvars index 130d1b3bf2f0..bebf9cedafa2 100644 --- a/infra/tpu-pytorch-releases/artifacts.auto.tfvars +++ b/infra/tpu-pytorch-releases/artifacts.auto.tfvars @@ -61,6 +61,7 @@ versioned_builds = [ package_version = "2.1.0rc2+libtpu" accelerator = "tpu" python_version = "3.10" + bundle_libtpu = "1" }, { git_tag = "v2.0.0" From b753640303feaa40a12cfb69d55b5e11f3d1ac87 Mon Sep 17 00:00:00 2001 From: Will Cromar Date: Wed, 13 Sep 2023 23:06:08 +0000 Subject: [PATCH 3/3] sanitize package version --- infra/tpu-pytorch-releases/artifacts_builds.tf | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/infra/tpu-pytorch-releases/artifacts_builds.tf b/infra/tpu-pytorch-releases/artifacts_builds.tf index 3e6aa2aa9c62..cba72bfe8b72 100644 --- a/infra/tpu-pytorch-releases/artifacts_builds.tf +++ b/infra/tpu-pytorch-releases/artifacts_builds.tf @@ -69,7 +69,7 @@ locals { versioned_builds_dict = { for b in var.versioned_builds : format("r%s_%s_%s", - b.package_version, + replace(b.package_version, "+", "_"), b.python_version, b.accelerator == "tpu" ? "tpuvm" : format("cuda_%s", b.cuda_version) ) => b