Skip to content

Commit 51df39f

Browse files
authored
Update artifacts.auto.tfvars for 2.2 release (#5908)
* add release build for v2.2 for xl-ml test * remove 2.2 xrt build * update release tag name * update package version name
1 parent f06f6c0 commit 51df39f

File tree

1 file changed

+73
-16
lines changed

1 file changed

+73
-16
lines changed

infra/tpu-pytorch-releases/artifacts.auto.tfvars

Lines changed: 73 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -38,26 +38,44 @@ nightly_builds = [
3838
}
3939
]
4040

41-
# TODO: Remove this after the 2.1 release
42-
xrt_versioned_builds = [
41+
# Built on push to specific tag.
42+
versioned_builds = [
43+
# Remove libtpu from PyPI builds
4344
{
44-
accelerator = "tpu"
45-
python_version = "3.10"
46-
pytorch_git_rev = "v2.1.0"
47-
package_version = "2.1.0+xrt"
45+
git_tag = "v2.2.0-rc1"
46+
package_version = "2.2.0-rc1"
47+
accelerator = "tpu"
48+
bundle_libtpu = "0"
4849
},
4950
{
50-
accelerator = "cuda"
51-
python_version = "3.10"
52-
cuda_version = "12.0"
53-
pytorch_git_rev = "v2.1.0"
54-
package_version = "2.1.0+xrt"
51+
git_tag = "v2.2.0-rc1"
52+
package_version = "2.2.0-rc1"
53+
accelerator = "tpu"
54+
python_version = "3.9"
55+
bundle_libtpu = "0"
56+
},
57+
{
58+
git_tag = "v2.2.0-rc1"
59+
package_version = "2.2.0-rc1"
60+
accelerator = "tpu"
61+
python_version = "3.10"
62+
bundle_libtpu = "0"
63+
},
64+
{
65+
git_tag = "v2.2.0-rc1"
66+
package_version = "2.2.0-rc1"
67+
accelerator = "tpu"
68+
python_version = "3.11"
69+
bundle_libtpu = "0"
70+
},
71+
# Bundle libtpu for Kaggle
72+
{
73+
git_tag = "v2.2.0-rc1"
74+
package_version = "2.2.0-rc1+libtpu"
75+
accelerator = "tpu"
76+
python_version = "3.10"
77+
bundle_libtpu = "1"
5578
},
56-
]
57-
58-
# Built on push to specific tag.
59-
versioned_builds = [
60-
# Remove libtpu from PyPI builds
6179
{
6280
git_tag = "v2.1.0"
6381
pytorch_git_rev = "v2.1.0"
@@ -108,6 +126,45 @@ versioned_builds = [
108126
package_version = "1.13"
109127
accelerator = "tpu"
110128
},
129+
{
130+
git_tag = "v2.2.0-rc1"
131+
package_version = "2.2.0-rc1"
132+
accelerator = "cuda"
133+
cuda_version = "11.8"
134+
},
135+
{
136+
git_tag = "v2.2.0-rc1"
137+
package_version = "2.2.0-rc1"
138+
accelerator = "cuda"
139+
cuda_version = "12.0"
140+
},
141+
{
142+
git_tag = "v2.2.0-rc1"
143+
package_version = "2.2.0-rc1"
144+
accelerator = "cuda"
145+
cuda_version = "12.1"
146+
},
147+
{
148+
git_tag = "v2.2.0-rc1"
149+
package_version = "2.2.0-rc1"
150+
accelerator = "cuda"
151+
cuda_version = "11.8"
152+
python_version = "3.10"
153+
},
154+
{
155+
git_tag = "v2.2.0-rc1"
156+
package_version = "2.2.0-rc1"
157+
accelerator = "cuda"
158+
cuda_version = "12.0"
159+
python_version = "3.10"
160+
},
161+
{
162+
git_tag = "v2.2.0-rc1"
163+
package_version = "2.2.0-rc1"
164+
accelerator = "cuda"
165+
cuda_version = "12.1"
166+
python_version = "3.10"
167+
},
111168
{
112169
git_tag = "v2.1.0"
113170
pytorch_git_rev = "v2.1.0"

0 commit comments

Comments
 (0)