Skip to content

Commit c395570

Browse files
gcf-owl-bot[bot]gcf-owl-bot[bot]
authored
feat: [tpu] Add ResetQueueResource to the TPU service (#4577)
* feat: Add ResetQueueResource to the TPU service feat: Add MultiNodeParams in TPU PiperOrigin-RevId: 559244735 Source-Link: googleapis/googleapis@75a3251 Source-Link: googleapis/googleapis-gen@b254470 Copy-Tag: eyJwIjoicGFja2FnZXMvZ29vZ2xlLWNsb3VkLXRwdS8uT3dsQm90LnlhbWwiLCJoIjoiYjI1NDQ3MDVhMjEwOTBmMzZiM2UyZjAwYTdjNmM0NzAwMDA4MTU3MyJ9 * 🦉 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>
1 parent ecfe8f8 commit c395570

9 files changed

Lines changed: 1229 additions & 0 deletions

File tree

‎packages/google-cloud-tpu/protos/google/cloud/tpu/v2alpha1/cloud_tpu.proto‎

Lines changed: 47 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -164,6 +164,20 @@ service Tpu {
164164
};
165165
}
166166

167+
// Resets a QueuedResource TPU instance
168+
rpc ResetQueuedResource(ResetQueuedResourceRequest)
169+
returns (google.longrunning.Operation) {
170+
option (google.api.http) = {
171+
post: "/v2alpha1/{name=projects/*/locations/*/queuedResources/*}:reset"
172+
body: "*"
173+
};
174+
option (google.api.method_signature) = "name";
175+
option (google.longrunning.operation_info) = {
176+
response_type: "QueuedResource"
177+
metadata_type: "OperationMetadata"
178+
};
179+
}
180+
167181
// Generates the Cloud TPU service identity for the project.
168182
rpc GenerateServiceIdentity(GenerateServiceIdentityRequest)
169183
returns (GenerateServiceIdentityResponse) {
@@ -544,6 +558,23 @@ message QueuedResource {
544558
// single node or multiple nodes.
545559
// NodeSpec provides the specification for node(s) to be created.
546560
message NodeSpec {
561+
// Parameters to specify for multi-node QueuedResource requests. This
562+
// field must be populated in case of multi-node requests instead of
563+
// node_id. It's an error to specify both node_id and multi_node_params.
564+
message MultiNodeParams {
565+
// Required. Number of nodes with this spec. The system will attempt
566+
// to provison "node_count" nodes as part of the request.
567+
// This needs to be > 1.
568+
int32 node_count = 1 [(google.api.field_behavior) = REQUIRED];
569+
570+
// Prefix of node_ids in case of multi-node request
571+
// Should follow the `^[A-Za-z0-9_.~+%-]+$` regex format.
572+
// If node_count = 3 and node_id_prefix = "np", node ids of nodes
573+
// created will be "np-0", "np-1", "np-2". If this field is not
574+
// provided we use queued_resource_id as the node_id_prefix.
575+
string node_id_prefix = 2;
576+
}
577+
547578
// Required. The parent resource name.
548579
string parent = 1 [
549580
(google.api.field_behavior) = REQUIRED,
@@ -558,6 +589,10 @@ message QueuedResource {
558589
// instead. It's an error to specify both node_id and multi_node_params.
559590
string node_id = 2;
560591

592+
// Optional. Fields to specify in case of multi-node request.
593+
MultiNodeParams multi_node_params = 6
594+
[(google.api.field_behavior) = OPTIONAL];
595+
561596
// Required. The node.
562597
Node node = 3 [(google.api.field_behavior) = REQUIRED];
563598
}
@@ -946,6 +981,18 @@ message DeleteQueuedResourceRequest {
946981
bool force = 3;
947982
}
948983

984+
// Request for
985+
// [ResetQueuedResource][google.cloud.tpu.v2alpha1.Tpu.ResetQueuedResource].
986+
message ResetQueuedResourceRequest {
987+
// Required. The name of the queued resource.
988+
string name = 1 [
989+
(google.api.field_behavior) = REQUIRED,
990+
(google.api.resource_reference) = {
991+
type: "tpu.googleapis.com/QueuedResource"
992+
}
993+
];
994+
}
995+
949996
// The per-product per-project service identity for Cloud TPU service.
950997
message ServiceIdentity {
951998
// The email address of the service identity.

‎packages/google-cloud-tpu/protos/protos.d.ts‎

Lines changed: 230 additions & 0 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

‎packages/google-cloud-tpu/protos/protos.js‎

Lines changed: 491 additions & 0 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

‎packages/google-cloud-tpu/protos/protos.json‎

Lines changed: 64 additions & 0 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

‎packages/google-cloud-tpu/samples/generated/v2alpha1/snippet_metadata_google.cloud.tpu.v2alpha1.json‎

Lines changed: 40 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -507,6 +507,46 @@
507507
}
508508
}
509509
},
510+
{
511+
"regionTag": "tpu_v2alpha1_generated_Tpu_ResetQueuedResource_async",
512+
"title": "Tpu resetQueuedResource Sample",
513+
"origin": "API_DEFINITION",
514+
"description": " Resets a QueuedResource TPU instance",
515+
"canonical": true,
516+
"file": "tpu.reset_queued_resource.js",
517+
"language": "JAVASCRIPT",
518+
"segments": [
519+
{
520+
"start": 25,
521+
"end": 54,
522+
"type": "FULL"
523+
}
524+
],
525+
"clientMethod": {
526+
"shortName": "ResetQueuedResource",
527+
"fullName": "google.cloud.tpu.v2alpha1.Tpu.ResetQueuedResource",
528+
"async": true,
529+
"parameters": [
530+
{
531+
"name": "name",
532+
"type": "TYPE_STRING"
533+
}
534+
],
535+
"resultType": ".google.longrunning.Operation",
536+
"client": {
537+
"shortName": "TpuClient",
538+
"fullName": "google.cloud.tpu.v2alpha1.TpuClient"
539+
},
540+
"method": {
541+
"shortName": "ResetQueuedResource",
542+
"fullName": "google.cloud.tpu.v2alpha1.Tpu.ResetQueuedResource",
543+
"service": {
544+
"shortName": "Tpu",
545+
"fullName": "google.cloud.tpu.v2alpha1.Tpu"
546+
}
547+
}
548+
}
549+
},
510550
{
511551
"regionTag": "tpu_v2alpha1_generated_Tpu_GenerateServiceIdentity_async",
512552
"title": "Tpu generateServiceIdentity Sample",

‎packages/google-cloud-tpu/src/v2alpha1/gapic_metadata.json‎

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -75,6 +75,11 @@
7575
"deleteQueuedResource"
7676
]
7777
},
78+
"ResetQueuedResource": {
79+
"methods": [
80+
"resetQueuedResource"
81+
]
82+
},
7883
"SimulateMaintenanceEvent": {
7984
"methods": [
8085
"simulateMaintenanceEvent"
@@ -178,6 +183,11 @@
178183
"deleteQueuedResource"
179184
]
180185
},
186+
"ResetQueuedResource": {
187+
"methods": [
188+
"resetQueuedResource"
189+
]
190+
},
181191
"SimulateMaintenanceEvent": {
182192
"methods": [
183193
"simulateMaintenanceEvent"

0 commit comments

Comments
 (0)