@@ -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