Skip to content

Commit

Permalink
feat: [tpu] Add spot tier in v2alpha Create QueuedResource API requ…
Browse files Browse the repository at this point in the history 
…est (#9686)

* feat: enable QueuedResources in us-east5
feat: add multislice_node flag

PiperOrigin-RevId: 550674209

Source-Link: googleapis/googleapis@2ad3b4c

Source-Link: googleapis/googleapis-gen@dbaaa6f
Copy-Tag: eyJwIjoiamF2YS10cHUvLk93bEJvdC55YW1sIiwiaCI6ImRiYWFhNmYxYzFkYjliOWI0YjkzZjc1NWM1YzAyOTEyODY1MDY2YWQifQ==

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* feat: Add `spot` tier in v2alpha Create QueuedResource API request

PiperOrigin-RevId: 551066097

Source-Link: googleapis/googleapis@b9e70ad

Source-Link: googleapis/googleapis-gen@b41d4d5
Copy-Tag: eyJwIjoiamF2YS10cHUvLk93bEJvdC55YW1sIiwiaCI6ImI0MWQ0ZDUyY2ZkNmUyYTBkODVjYWYyN2M5NzhiM2Q3ZjAyNDQ4ZDgifQ==

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

---------

Co-authored-by: Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>
  • Loading branch information
@gcf-owl-bot
gcf-owl-bot[bot] and gcf-owl-bot[bot] committed Aug 2, 2023
1 parent 205f373 commit cbf3dd5
Show file tree
Hide file tree
Showing 14 changed files with 1,666 additions and 482 deletions.
2 changes: 1 addition & 1 deletion java-tpu/README.md
View file
Expand Up @@ -195,7 +195,7 @@ Java is a registered trademark of Oracle and/or its affiliates.
[kokoro-badge-link-5]: http://storage.googleapis.com/cloud-devrel-public/java/badges/google-cloud-java/java11.html
[stability-image]: https://img.shields.io/badge/stability-stable-green
[maven-version-image]: https://img.shields.io/maven-central/v/com.google.cloud/google-cloud-tpu.svg
[maven-version-link]: https://central.sonatype.com/artifact/com.google.cloud/google-cloud-tpu/2.22.0
[maven-version-link]: https://central.sonatype.com/artifact/com.google.cloud/google-cloud-tpu/2.23.0
[authentication]: https://github.com/googleapis/google-cloud-java#authentication
[auth-scopes]: https://developers.google.com/identity/protocols/oauth2/scopes
[predefined-iam-roles]: https://cloud.google.com/iam/docs/understanding-roles#predefined_roles
Expand Down
18 changes: 18 additions & 0 deletions ...rc/main/resources/META-INF/native-image/com.google.cloud.tpu.v2alpha1/reflect-config.json
View file
Expand Up @@ -1124,6 +1124,24 @@
"allDeclaredClasses": true,
"allPublicClasses": true
},
{
"name": "com.google.cloud.tpu.v2alpha1.QueuedResource$Spot",
"queryAllDeclaredConstructors": true,
"queryAllPublicConstructors": true,
"queryAllDeclaredMethods": true,
"allPublicMethods": true,
"allDeclaredClasses": true,
"allPublicClasses": true
},
{
"name": "com.google.cloud.tpu.v2alpha1.QueuedResource$Spot$Builder",
"queryAllDeclaredConstructors": true,
"queryAllPublicConstructors": true,
"queryAllDeclaredMethods": true,
"allPublicMethods": true,
"allDeclaredClasses": true,
"allPublicClasses": true
},
{
"name": "com.google.cloud.tpu.v2alpha1.QueuedResource$Tpu",
"queryAllDeclaredConstructors": true,
Expand Down
18 changes: 18 additions & 0 deletions java-tpu/google-cloud-tpu/src/test/java/com/google/cloud/tpu/v2/TpuClientHttpJsonTest.java
View file
Expand Up @@ -212,6 +212,8 @@ public void getNodeTest() throws Exception {
.addAllSymptoms(new ArrayList<Symptom>())
.setShieldedInstanceConfig(ShieldedInstanceConfig.newBuilder().build())
.setAcceleratorConfig(AcceleratorConfig.newBuilder().build())
.setQueuedResource("queuedResource94310049")
.setMultisliceNode(true)
.build();
mockService.addResponse(expectedResponse);

Expand Down Expand Up @@ -274,6 +276,8 @@ public void getNodeTest2() throws Exception {
.addAllSymptoms(new ArrayList<Symptom>())
.setShieldedInstanceConfig(ShieldedInstanceConfig.newBuilder().build())
.setAcceleratorConfig(AcceleratorConfig.newBuilder().build())
.setQueuedResource("queuedResource94310049")
.setMultisliceNode(true)
.build();
mockService.addResponse(expectedResponse);

Expand Down Expand Up @@ -336,6 +340,8 @@ public void createNodeTest() throws Exception {
.addAllSymptoms(new ArrayList<Symptom>())
.setShieldedInstanceConfig(ShieldedInstanceConfig.newBuilder().build())
.setAcceleratorConfig(AcceleratorConfig.newBuilder().build())
.setQueuedResource("queuedResource94310049")
.setMultisliceNode(true)
.build();
Operation resultOperation =
Operation.newBuilder()
Expand Down Expand Up @@ -407,6 +413,8 @@ public void createNodeTest2() throws Exception {
.addAllSymptoms(new ArrayList<Symptom>())
.setShieldedInstanceConfig(ShieldedInstanceConfig.newBuilder().build())
.setAcceleratorConfig(AcceleratorConfig.newBuilder().build())
.setQueuedResource("queuedResource94310049")
.setMultisliceNode(true)
.build();
Operation resultOperation =
Operation.newBuilder()
Expand Down Expand Up @@ -568,6 +576,8 @@ public void stopNodeTest() throws Exception {
.addAllSymptoms(new ArrayList<Symptom>())
.setShieldedInstanceConfig(ShieldedInstanceConfig.newBuilder().build())
.setAcceleratorConfig(AcceleratorConfig.newBuilder().build())
.setQueuedResource("queuedResource94310049")
.setMultisliceNode(true)
.build();
Operation resultOperation =
Operation.newBuilder()
Expand Down Expand Up @@ -641,6 +651,8 @@ public void startNodeTest() throws Exception {
.addAllSymptoms(new ArrayList<Symptom>())
.setShieldedInstanceConfig(ShieldedInstanceConfig.newBuilder().build())
.setAcceleratorConfig(AcceleratorConfig.newBuilder().build())
.setQueuedResource("queuedResource94310049")
.setMultisliceNode(true)
.build();
Operation resultOperation =
Operation.newBuilder()
Expand Down Expand Up @@ -714,6 +726,8 @@ public void updateNodeTest() throws Exception {
.addAllSymptoms(new ArrayList<Symptom>())
.setShieldedInstanceConfig(ShieldedInstanceConfig.newBuilder().build())
.setAcceleratorConfig(AcceleratorConfig.newBuilder().build())
.setQueuedResource("queuedResource94310049")
.setMultisliceNode(true)
.build();
Operation resultOperation =
Operation.newBuilder()
Expand Down Expand Up @@ -744,6 +758,8 @@ public void updateNodeTest() throws Exception {
.addAllSymptoms(new ArrayList<Symptom>())
.setShieldedInstanceConfig(ShieldedInstanceConfig.newBuilder().build())
.setAcceleratorConfig(AcceleratorConfig.newBuilder().build())
.setQueuedResource("queuedResource94310049")
.setMultisliceNode(true)
.build();
FieldMask updateMask = FieldMask.newBuilder().build();

Expand Down Expand Up @@ -794,6 +810,8 @@ public void updateNodeExceptionTest() throws Exception {
.addAllSymptoms(new ArrayList<Symptom>())
.setShieldedInstanceConfig(ShieldedInstanceConfig.newBuilder().build())
.setAcceleratorConfig(AcceleratorConfig.newBuilder().build())
.setQueuedResource("queuedResource94310049")
.setMultisliceNode(true)
.build();
FieldMask updateMask = FieldMask.newBuilder().build();
client.updateNodeAsync(node, updateMask).get();
Expand Down
14 changes: 14 additions & 0 deletions java-tpu/google-cloud-tpu/src/test/java/com/google/cloud/tpu/v2/TpuClientTest.java
View file
Expand Up @@ -207,6 +207,8 @@ public void getNodeTest() throws Exception {
.addAllSymptoms(new ArrayList<Symptom>())
.setShieldedInstanceConfig(ShieldedInstanceConfig.newBuilder().build())
.setAcceleratorConfig(AcceleratorConfig.newBuilder().build())
.setQueuedResource("queuedResource94310049")
.setMultisliceNode(true)
.build();
mockTpu.addResponse(expectedResponse);

Expand Down Expand Up @@ -263,6 +265,8 @@ public void getNodeTest2() throws Exception {
.addAllSymptoms(new ArrayList<Symptom>())
.setShieldedInstanceConfig(ShieldedInstanceConfig.newBuilder().build())
.setAcceleratorConfig(AcceleratorConfig.newBuilder().build())
.setQueuedResource("queuedResource94310049")
.setMultisliceNode(true)
.build();
mockTpu.addResponse(expectedResponse);

Expand Down Expand Up @@ -319,6 +323,8 @@ public void createNodeTest() throws Exception {
.addAllSymptoms(new ArrayList<Symptom>())
.setShieldedInstanceConfig(ShieldedInstanceConfig.newBuilder().build())
.setAcceleratorConfig(AcceleratorConfig.newBuilder().build())
.setQueuedResource("queuedResource94310049")
.setMultisliceNode(true)
.build();
Operation resultOperation =
Operation.newBuilder()
Expand Down Expand Up @@ -389,6 +395,8 @@ public void createNodeTest2() throws Exception {
.addAllSymptoms(new ArrayList<Symptom>())
.setShieldedInstanceConfig(ShieldedInstanceConfig.newBuilder().build())
.setAcceleratorConfig(AcceleratorConfig.newBuilder().build())
.setQueuedResource("queuedResource94310049")
.setMultisliceNode(true)
.build();
Operation resultOperation =
Operation.newBuilder()
Expand Down Expand Up @@ -543,6 +551,8 @@ public void stopNodeTest() throws Exception {
.addAllSymptoms(new ArrayList<Symptom>())
.setShieldedInstanceConfig(ShieldedInstanceConfig.newBuilder().build())
.setAcceleratorConfig(AcceleratorConfig.newBuilder().build())
.setQueuedResource("queuedResource94310049")
.setMultisliceNode(true)
.build();
Operation resultOperation =
Operation.newBuilder()
Expand Down Expand Up @@ -613,6 +623,8 @@ public void startNodeTest() throws Exception {
.addAllSymptoms(new ArrayList<Symptom>())
.setShieldedInstanceConfig(ShieldedInstanceConfig.newBuilder().build())
.setAcceleratorConfig(AcceleratorConfig.newBuilder().build())
.setQueuedResource("queuedResource94310049")
.setMultisliceNode(true)
.build();
Operation resultOperation =
Operation.newBuilder()
Expand Down Expand Up @@ -683,6 +695,8 @@ public void updateNodeTest() throws Exception {
.addAllSymptoms(new ArrayList<Symptom>())
.setShieldedInstanceConfig(ShieldedInstanceConfig.newBuilder().build())
.setAcceleratorConfig(AcceleratorConfig.newBuilder().build())
.setQueuedResource("queuedResource94310049")
.setMultisliceNode(true)
.build();
Operation resultOperation =
Operation.newBuilder()
Expand Down
14 changes: 7 additions & 7 deletions ...-tpu/proto-google-cloud-tpu-v2/src/main/java/com/google/cloud/tpu/v2/AcceleratorType.java
View file
Expand Up @@ -123,7 +123,7 @@ public com.google.protobuf.ByteString getNameBytes() {
*
*
* <pre>
* the accelerator type.
* The accelerator type.
* </pre>
*
* <code>string type = 2;</code>
Expand All @@ -146,7 +146,7 @@ public java.lang.String getType() {
*
*
* <pre>
* the accelerator type.
* The accelerator type.
* </pre>
*
* <code>string type = 2;</code>
Expand Down Expand Up @@ -785,7 +785,7 @@ public Builder setNameBytes(com.google.protobuf.ByteString value) {
*
*
* <pre>
* the accelerator type.
* The accelerator type.
* </pre>
*
* <code>string type = 2;</code>
Expand All @@ -807,7 +807,7 @@ public java.lang.String getType() {
*
*
* <pre>
* the accelerator type.
* The accelerator type.
* </pre>
*
* <code>string type = 2;</code>
Expand All @@ -829,7 +829,7 @@ public com.google.protobuf.ByteString getTypeBytes() {
*
*
* <pre>
* the accelerator type.
* The accelerator type.
* </pre>
*
* <code>string type = 2;</code>
Expand All @@ -850,7 +850,7 @@ public Builder setType(java.lang.String value) {
*
*
* <pre>
* the accelerator type.
* The accelerator type.
* </pre>
*
* <code>string type = 2;</code>
Expand All @@ -867,7 +867,7 @@ public Builder clearType() {
*
*
* <pre>
* the accelerator type.
* The accelerator type.
* </pre>
*
* <code>string type = 2;</code>
Expand Down
4 changes: 2 additions & 2 deletions ...o-google-cloud-tpu-v2/src/main/java/com/google/cloud/tpu/v2/AcceleratorTypeOrBuilder.java
View file
Expand Up @@ -52,7 +52,7 @@ public interface AcceleratorTypeOrBuilder
*
*
* <pre>
* the accelerator type.
* The accelerator type.
* </pre>
*
* <code>string type = 2;</code>
Expand All @@ -64,7 +64,7 @@ public interface AcceleratorTypeOrBuilder
*
*
* <pre>
* the accelerator type.
* The accelerator type.
* </pre>
*
* <code>string type = 2;</code>
Expand Down

0 comments on commit cbf3dd5

Please sign in to comment.