Skip to content

Commit 9996b84

Browse files
gcf-owl-bot[bot]gcf-owl-bot[bot]
authored
feat: [tpu] Add ResetQueueResource to the TPU service (#9748)
- [ ] Regenerate this pull request now. feat: Add MultiNodeParams in TPU PiperOrigin-RevId: 559244735 Source-Link: https://togithub.com/googleapis/googleapis/commit/75a3251882cca05096fe3a620eb67c0c52f45c8b Source-Link: https://togithub.com/googleapis/googleapis-gen/commit/b2544705a21090f36b3e2f00a7c6c47000081573 Copy-Tag: eyJwIjoiamF2YS10cHUvLk93bEJvdC55YW1sIiwiaCI6ImIyNTQ0NzA1YTIxMDkwZjM2YjNlMmYwMGE3YzZjNDcwMDAwODE1NzMifQ== BEGIN_NESTED_COMMIT build: [tpu] do not create bazel-* symlinks PiperOrigin-RevId: 555307834 Source-Link: https://togithub.com/googleapis/googleapis/commit/43f02766add4ac5e2e6557a9d2600f2101654dff Source-Link: https://togithub.com/googleapis/googleapis-gen/commit/f340c3f9408f87a93ab676233181b2738de4a9fe Copy-Tag: eyJwIjoiamF2YS10cHUvLk93bEJvdC55YW1sIiwiaCI6ImYzNDBjM2Y5NDA4Zjg3YTkzYWI2NzYyMzMxODFiMjczOGRlNGE5ZmUifQ== END_NESTED_COMMIT
1 parent 48ac78c commit 9996b84

File tree

23 files changed

+3473
-300
lines changed

23 files changed

+3473
-300
lines changed

‎java-tpu/README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ If you are using Maven with [BOM][libraries-bom], add this to your pom.xml file:
2020
<dependency>
2121
<groupId>com.google.cloud</groupId>
2222
<artifactId>libraries-bom</artifactId>
23-
<version>26.19.0</version>
23+
<version>26.22.0</version>
2424
<type>pom</type>
2525
<scope>import</scope>
2626
</dependency>
@@ -195,7 +195,7 @@ Java is a registered trademark of Oracle and/or its affiliates.
195195
[kokoro-badge-link-5]: http://storage.googleapis.com/cloud-devrel-public/java/badges/google-cloud-java/java11.html
196196
[stability-image]: https://img.shields.io/badge/stability-stable-green
197197
[maven-version-image]: https://img.shields.io/maven-central/v/com.google.cloud/google-cloud-tpu.svg
198-
[maven-version-link]: https://central.sonatype.com/artifact/com.google.cloud/google-cloud-tpu/2.23.0
198+
[maven-version-link]: https://central.sonatype.com/artifact/com.google.cloud/google-cloud-tpu/2.24.0
199199
[authentication]: https://github.com/googleapis/google-cloud-java#authentication
200200
[auth-scopes]: https://developers.google.com/identity/protocols/oauth2/scopes
201201
[predefined-iam-roles]: https://cloud.google.com/iam/docs/understanding-roles#predefined_roles

‎java-tpu/google-cloud-tpu/src/main/java/com/google/cloud/tpu/v2alpha1/TpuClient.java

Lines changed: 148 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1639,6 +1639,154 @@ public final OperationFuture<QueuedResource, OperationMetadata> deleteQueuedReso
16391639
return stub.deleteQueuedResourceCallable();
16401640
}
16411641

1642+
// AUTO-GENERATED DOCUMENTATION AND METHOD.
1643+
/**
1644+
* Resets a QueuedResource TPU instance
1645+
*
1646+
* <p>Sample code:
1647+
*
1648+
* <pre>{@code
1649+
* // This snippet has been automatically generated and should be regarded as a code template only.
1650+
* // It will require modifications to work:
1651+
* // - It may require correct/in-range values for request initialization.
1652+
* // - It may require specifying regional endpoints when creating the service client as shown in
1653+
* // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
1654+
* try (TpuClient tpuClient = TpuClient.create()) {
1655+
* QueuedResourceName name =
1656+
* QueuedResourceName.of("[PROJECT]", "[LOCATION]", "[QUEUED_RESOURCE]");
1657+
* QueuedResource response = tpuClient.resetQueuedResourceAsync(name).get();
1658+
* }
1659+
* }</pre>
1660+
*
1661+
* @param name Required. The name of the queued resource.
1662+
* @throws com.google.api.gax.rpc.ApiException if the remote call fails
1663+
*/
1664+
public final OperationFuture<QueuedResource, OperationMetadata> resetQueuedResourceAsync(
1665+
QueuedResourceName name) {
1666+
ResetQueuedResourceRequest request =
1667+
ResetQueuedResourceRequest.newBuilder()
1668+
.setName(name == null ? null : name.toString())
1669+
.build();
1670+
return resetQueuedResourceAsync(request);
1671+
}
1672+
1673+
// AUTO-GENERATED DOCUMENTATION AND METHOD.
1674+
/**
1675+
* Resets a QueuedResource TPU instance
1676+
*
1677+
* <p>Sample code:
1678+
*
1679+
* <pre>{@code
1680+
* // This snippet has been automatically generated and should be regarded as a code template only.
1681+
* // It will require modifications to work:
1682+
* // - It may require correct/in-range values for request initialization.
1683+
* // - It may require specifying regional endpoints when creating the service client as shown in
1684+
* // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
1685+
* try (TpuClient tpuClient = TpuClient.create()) {
1686+
* String name =
1687+
* QueuedResourceName.of("[PROJECT]", "[LOCATION]", "[QUEUED_RESOURCE]").toString();
1688+
* QueuedResource response = tpuClient.resetQueuedResourceAsync(name).get();
1689+
* }
1690+
* }</pre>
1691+
*
1692+
* @param name Required. The name of the queued resource.
1693+
* @throws com.google.api.gax.rpc.ApiException if the remote call fails
1694+
*/
1695+
public final OperationFuture<QueuedResource, OperationMetadata> resetQueuedResourceAsync(
1696+
String name) {
1697+
ResetQueuedResourceRequest request =
1698+
ResetQueuedResourceRequest.newBuilder().setName(name).build();
1699+
return resetQueuedResourceAsync(request);
1700+
}
1701+
1702+
// AUTO-GENERATED DOCUMENTATION AND METHOD.
1703+
/**
1704+
* Resets a QueuedResource TPU instance
1705+
*
1706+
* <p>Sample code:
1707+
*
1708+
* <pre>{@code
1709+
* // This snippet has been automatically generated and should be regarded as a code template only.
1710+
* // It will require modifications to work:
1711+
* // - It may require correct/in-range values for request initialization.
1712+
* // - It may require specifying regional endpoints when creating the service client as shown in
1713+
* // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
1714+
* try (TpuClient tpuClient = TpuClient.create()) {
1715+
* ResetQueuedResourceRequest request =
1716+
* ResetQueuedResourceRequest.newBuilder()
1717+
* .setName(
1718+
* QueuedResourceName.of("[PROJECT]", "[LOCATION]", "[QUEUED_RESOURCE]").toString())
1719+
* .build();
1720+
* QueuedResource response = tpuClient.resetQueuedResourceAsync(request).get();
1721+
* }
1722+
* }</pre>
1723+
*
1724+
* @param request The request object containing all of the parameters for the API call.
1725+
* @throws com.google.api.gax.rpc.ApiException if the remote call fails
1726+
*/
1727+
public final OperationFuture<QueuedResource, OperationMetadata> resetQueuedResourceAsync(
1728+
ResetQueuedResourceRequest request) {
1729+
return resetQueuedResourceOperationCallable().futureCall(request);
1730+
}
1731+
1732+
// AUTO-GENERATED DOCUMENTATION AND METHOD.
1733+
/**
1734+
* Resets a QueuedResource TPU instance
1735+
*
1736+
* <p>Sample code:
1737+
*
1738+
* <pre>{@code
1739+
* // This snippet has been automatically generated and should be regarded as a code template only.
1740+
* // It will require modifications to work:
1741+
* // - It may require correct/in-range values for request initialization.
1742+
* // - It may require specifying regional endpoints when creating the service client as shown in
1743+
* // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
1744+
* try (TpuClient tpuClient = TpuClient.create()) {
1745+
* ResetQueuedResourceRequest request =
1746+
* ResetQueuedResourceRequest.newBuilder()
1747+
* .setName(
1748+
* QueuedResourceName.of("[PROJECT]", "[LOCATION]", "[QUEUED_RESOURCE]").toString())
1749+
* .build();
1750+
* OperationFuture<QueuedResource, OperationMetadata> future =
1751+
* tpuClient.resetQueuedResourceOperationCallable().futureCall(request);
1752+
* // Do something.
1753+
* QueuedResource response = future.get();
1754+
* }
1755+
* }</pre>
1756+
*/
1757+
public final OperationCallable<ResetQueuedResourceRequest, QueuedResource, OperationMetadata>
1758+
resetQueuedResourceOperationCallable() {
1759+
return stub.resetQueuedResourceOperationCallable();
1760+
}
1761+
1762+
// AUTO-GENERATED DOCUMENTATION AND METHOD.
1763+
/**
1764+
* Resets a QueuedResource TPU instance
1765+
*
1766+
* <p>Sample code:
1767+
*
1768+
* <pre>{@code
1769+
* // This snippet has been automatically generated and should be regarded as a code template only.
1770+
* // It will require modifications to work:
1771+
* // - It may require correct/in-range values for request initialization.
1772+
* // - It may require specifying regional endpoints when creating the service client as shown in
1773+
* // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
1774+
* try (TpuClient tpuClient = TpuClient.create()) {
1775+
* ResetQueuedResourceRequest request =
1776+
* ResetQueuedResourceRequest.newBuilder()
1777+
* .setName(
1778+
* QueuedResourceName.of("[PROJECT]", "[LOCATION]", "[QUEUED_RESOURCE]").toString())
1779+
* .build();
1780+
* ApiFuture<Operation> future = tpuClient.resetQueuedResourceCallable().futureCall(request);
1781+
* // Do something.
1782+
* Operation response = future.get();
1783+
* }
1784+
* }</pre>
1785+
*/
1786+
public final UnaryCallable<ResetQueuedResourceRequest, Operation> resetQueuedResourceCallable() {
1787+
return stub.resetQueuedResourceCallable();
1788+
}
1789+
16421790
// AUTO-GENERATED DOCUMENTATION AND METHOD.
16431791
/**
16441792
* Generates the Cloud TPU service identity for the project.

‎java-tpu/google-cloud-tpu/src/main/java/com/google/cloud/tpu/v2alpha1/TpuSettings.java

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -185,6 +185,17 @@ public UnaryCallSettings<DeleteQueuedResourceRequest, Operation> deleteQueuedRes
185185
return ((TpuStubSettings) getStubSettings()).deleteQueuedResourceOperationSettings();
186186
}
187187

188+
/** Returns the object with the settings used for calls to resetQueuedResource. */
189+
public UnaryCallSettings<ResetQueuedResourceRequest, Operation> resetQueuedResourceSettings() {
190+
return ((TpuStubSettings) getStubSettings()).resetQueuedResourceSettings();
191+
}
192+
193+
/** Returns the object with the settings used for calls to resetQueuedResource. */
194+
public OperationCallSettings<ResetQueuedResourceRequest, QueuedResource, OperationMetadata>
195+
resetQueuedResourceOperationSettings() {
196+
return ((TpuStubSettings) getStubSettings()).resetQueuedResourceOperationSettings();
197+
}
198+
188199
/** Returns the object with the settings used for calls to generateServiceIdentity. */
189200
public UnaryCallSettings<GenerateServiceIdentityRequest, GenerateServiceIdentityResponse>
190201
generateServiceIdentitySettings() {
@@ -450,6 +461,19 @@ public UnaryCallSettings.Builder<UpdateNodeRequest, Operation> updateNodeSetting
450461
return getStubSettingsBuilder().deleteQueuedResourceOperationSettings();
451462
}
452463

464+
/** Returns the builder for the settings used for calls to resetQueuedResource. */
465+
public UnaryCallSettings.Builder<ResetQueuedResourceRequest, Operation>
466+
resetQueuedResourceSettings() {
467+
return getStubSettingsBuilder().resetQueuedResourceSettings();
468+
}
469+
470+
/** Returns the builder for the settings used for calls to resetQueuedResource. */
471+
public OperationCallSettings.Builder<
472+
ResetQueuedResourceRequest, QueuedResource, OperationMetadata>
473+
resetQueuedResourceOperationSettings() {
474+
return getStubSettingsBuilder().resetQueuedResourceOperationSettings();
475+
}
476+
453477
/** Returns the builder for the settings used for calls to generateServiceIdentity. */
454478
public UnaryCallSettings.Builder<
455479
GenerateServiceIdentityRequest, GenerateServiceIdentityResponse>

‎java-tpu/google-cloud-tpu/src/main/java/com/google/cloud/tpu/v2alpha1/gapic_metadata.json

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -58,6 +58,9 @@
5858
"ListRuntimeVersions": {
5959
"methods": ["listRuntimeVersions", "listRuntimeVersions", "listRuntimeVersions", "listRuntimeVersionsPagedCallable", "listRuntimeVersionsCallable"]
6060
},
61+
"ResetQueuedResource": {
62+
"methods": ["resetQueuedResourceAsync", "resetQueuedResourceAsync", "resetQueuedResourceAsync", "resetQueuedResourceOperationCallable", "resetQueuedResourceCallable"]
63+
},
6164
"SimulateMaintenanceEvent": {
6265
"methods": ["simulateMaintenanceEventAsync", "simulateMaintenanceEventOperationCallable", "simulateMaintenanceEventCallable"]
6366
},

‎java-tpu/google-cloud-tpu/src/main/java/com/google/cloud/tpu/v2alpha1/stub/GrpcTpuStub.java

Lines changed: 46 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,7 @@
5959
import com.google.cloud.tpu.v2alpha1.Node;
6060
import com.google.cloud.tpu.v2alpha1.OperationMetadata;
6161
import com.google.cloud.tpu.v2alpha1.QueuedResource;
62+
import com.google.cloud.tpu.v2alpha1.ResetQueuedResourceRequest;
6263
import com.google.cloud.tpu.v2alpha1.RuntimeVersion;
6364
import com.google.cloud.tpu.v2alpha1.SimulateMaintenanceEventRequest;
6465
import com.google.cloud.tpu.v2alpha1.StartNodeRequest;
@@ -180,6 +181,16 @@ public class GrpcTpuStub extends TpuStub {
180181
.setResponseMarshaller(ProtoUtils.marshaller(Operation.getDefaultInstance()))
181182
.build();
182183

184+
private static final MethodDescriptor<ResetQueuedResourceRequest, Operation>
185+
resetQueuedResourceMethodDescriptor =
186+
MethodDescriptor.<ResetQueuedResourceRequest, Operation>newBuilder()
187+
.setType(MethodDescriptor.MethodType.UNARY)
188+
.setFullMethodName("google.cloud.tpu.v2alpha1.Tpu/ResetQueuedResource")
189+
.setRequestMarshaller(
190+
ProtoUtils.marshaller(ResetQueuedResourceRequest.getDefaultInstance()))
191+
.setResponseMarshaller(ProtoUtils.marshaller(Operation.getDefaultInstance()))
192+
.build();
193+
183194
private static final MethodDescriptor<
184195
GenerateServiceIdentityRequest, GenerateServiceIdentityResponse>
185196
generateServiceIdentityMethodDescriptor =
@@ -304,6 +315,9 @@ public class GrpcTpuStub extends TpuStub {
304315
private final UnaryCallable<DeleteQueuedResourceRequest, Operation> deleteQueuedResourceCallable;
305316
private final OperationCallable<DeleteQueuedResourceRequest, QueuedResource, OperationMetadata>
306317
deleteQueuedResourceOperationCallable;
318+
private final UnaryCallable<ResetQueuedResourceRequest, Operation> resetQueuedResourceCallable;
319+
private final OperationCallable<ResetQueuedResourceRequest, QueuedResource, OperationMetadata>
320+
resetQueuedResourceOperationCallable;
307321
private final UnaryCallable<GenerateServiceIdentityRequest, GenerateServiceIdentityResponse>
308322
generateServiceIdentityCallable;
309323
private final UnaryCallable<ListAcceleratorTypesRequest, ListAcceleratorTypesResponse>
@@ -476,6 +490,16 @@ protected GrpcTpuStub(
476490
return builder.build();
477491
})
478492
.build();
493+
GrpcCallSettings<ResetQueuedResourceRequest, Operation> resetQueuedResourceTransportSettings =
494+
GrpcCallSettings.<ResetQueuedResourceRequest, Operation>newBuilder()
495+
.setMethodDescriptor(resetQueuedResourceMethodDescriptor)
496+
.setParamsExtractor(
497+
request -> {
498+
RequestParamsBuilder builder = RequestParamsBuilder.create();
499+
builder.add("name", String.valueOf(request.getName()));
500+
return builder.build();
501+
})
502+
.build();
479503
GrpcCallSettings<GenerateServiceIdentityRequest, GenerateServiceIdentityResponse>
480504
generateServiceIdentityTransportSettings =
481505
GrpcCallSettings
@@ -665,6 +689,17 @@ protected GrpcTpuStub(
665689
settings.deleteQueuedResourceOperationSettings(),
666690
clientContext,
667691
operationsStub);
692+
this.resetQueuedResourceCallable =
693+
callableFactory.createUnaryCallable(
694+
resetQueuedResourceTransportSettings,
695+
settings.resetQueuedResourceSettings(),
696+
clientContext);
697+
this.resetQueuedResourceOperationCallable =
698+
callableFactory.createOperationCallable(
699+
resetQueuedResourceTransportSettings,
700+
settings.resetQueuedResourceOperationSettings(),
701+
clientContext,
702+
operationsStub);
668703
this.generateServiceIdentityCallable =
669704
callableFactory.createUnaryCallable(
670705
generateServiceIdentityTransportSettings,
@@ -841,6 +876,17 @@ public UnaryCallable<DeleteQueuedResourceRequest, Operation> deleteQueuedResourc
841876
return deleteQueuedResourceOperationCallable;
842877
}
843878

879+
@Override
880+
public UnaryCallable<ResetQueuedResourceRequest, Operation> resetQueuedResourceCallable() {
881+
return resetQueuedResourceCallable;
882+
}
883+
884+
@Override
885+
public OperationCallable<ResetQueuedResourceRequest, QueuedResource, OperationMetadata>
886+
resetQueuedResourceOperationCallable() {
887+
return resetQueuedResourceOperationCallable;
888+
}
889+
844890
@Override
845891
public UnaryCallable<GenerateServiceIdentityRequest, GenerateServiceIdentityResponse>
846892
generateServiceIdentityCallable() {

‎java-tpu/google-cloud-tpu/src/main/java/com/google/cloud/tpu/v2alpha1/stub/TpuStub.java

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -54,6 +54,7 @@
5454
import com.google.cloud.tpu.v2alpha1.Node;
5555
import com.google.cloud.tpu.v2alpha1.OperationMetadata;
5656
import com.google.cloud.tpu.v2alpha1.QueuedResource;
57+
import com.google.cloud.tpu.v2alpha1.ResetQueuedResourceRequest;
5758
import com.google.cloud.tpu.v2alpha1.RuntimeVersion;
5859
import com.google.cloud.tpu.v2alpha1.SimulateMaintenanceEventRequest;
5960
import com.google.cloud.tpu.v2alpha1.StartNodeRequest;
@@ -167,6 +168,16 @@ public UnaryCallable<DeleteQueuedResourceRequest, Operation> deleteQueuedResourc
167168
throw new UnsupportedOperationException("Not implemented: deleteQueuedResourceCallable()");
168169
}
169170

171+
public OperationCallable<ResetQueuedResourceRequest, QueuedResource, OperationMetadata>
172+
resetQueuedResourceOperationCallable() {
173+
throw new UnsupportedOperationException(
174+
"Not implemented: resetQueuedResourceOperationCallable()");
175+
}
176+
177+
public UnaryCallable<ResetQueuedResourceRequest, Operation> resetQueuedResourceCallable() {
178+
throw new UnsupportedOperationException("Not implemented: resetQueuedResourceCallable()");
179+
}
180+
170181
public UnaryCallable<GenerateServiceIdentityRequest, GenerateServiceIdentityResponse>
171182
generateServiceIdentityCallable() {
172183
throw new UnsupportedOperationException("Not implemented: generateServiceIdentityCallable()");

0 commit comments

Comments
 (0)