diff --git a/infra/tpu-pytorch-releases/artifacts.auto.tfvars b/infra/tpu-pytorch-releases/artifacts.auto.tfvars index bd5edb6d02af..cbec743e3806 100644 --- a/infra/tpu-pytorch-releases/artifacts.auto.tfvars +++ b/infra/tpu-pytorch-releases/artifacts.auto.tfvars @@ -35,14 +35,14 @@ xrt_versioned_builds = [ { accelerator = "tpu" python_version = "3.10" - pytorch_git_rev = "v2.1.0-rc6" + pytorch_git_rev = "v2.1.0" package_version = "2.1.0+xrt" }, { accelerator = "cuda" python_version = "3.10" cuda_version = "12.0" - pytorch_git_rev = "v2.1.0-rc6" + pytorch_git_rev = "v2.1.0" package_version = "2.1.0+xrt" }, ] @@ -51,14 +51,14 @@ xrt_versioned_builds = [ versioned_builds = [ { git_tag = "v2.1.0" - pytorch_git_rev = "v2.1.0-rc6" + pytorch_git_rev = "v2.1.0" package_version = "2.1.0" accelerator = "tpu" bundle_libtpu = "0" }, { git_tag = "v2.1.0" - pytorch_git_rev = "v2.1.0-rc6" + pytorch_git_rev = "v2.1.0" package_version = "2.1.0" accelerator = "tpu" python_version = "3.10" @@ -66,7 +66,7 @@ versioned_builds = [ }, { git_tag = "v2.1.0" - pytorch_git_rev = "v2.1.0-rc6" + pytorch_git_rev = "v2.1.0" package_version = "2.1.0+libtpu" accelerator = "tpu" python_version = "3.10" @@ -84,26 +84,41 @@ versioned_builds = [ }, { git_tag = "v2.1.0" - pytorch_git_rev = "v2.1.0-rc6" + pytorch_git_rev = "v2.1.0" package_version = "2.1.0", accelerator = "cuda" cuda_version = "12.0" }, { git_tag = "v2.1.0" - pytorch_git_rev = "v2.1.0-rc6" + pytorch_git_rev = "v2.1.0" package_version = "2.1.0" accelerator = "cuda" cuda_version = "11.8" }, { git_tag = "v2.1.0" - pytorch_git_rev = "v2.1.0-rc6" + pytorch_git_rev = "v2.1.0" + package_version = "2.1.0" + accelerator = "cuda" + cuda_version = "12.1" + }, + { + git_tag = "v2.1.0" + pytorch_git_rev = "v2.1.0" package_version = "2.1.0" accelerator = "cuda" cuda_version = "11.8" python_version = "3.10" }, + { + git_tag = "v2.1.0" + pytorch_git_rev = "v2.1.0" + package_version = "2.1.0" + accelerator = "cuda" + cuda_version = "12.1" + python_version = "3.10" + }, { git_tag = "v2.0.0" package_version = "2.0"