Skip to content

Commit 5c0a386

Browse files
yoshi-automationsofisl
authored andcommitted
feat(tpu): update the API
#### tpu:v1alpha1 The following keys were changed: - resources.projects.resources.locations.resources.operations.methods.cancel.description - schemas.Node.properties.state.enum - schemas.Node.properties.state.enumDescriptions #### tpu:v2alpha1 The following keys were added: - resources.projects.resources.locations.resources.nodes.methods.performMaintenance.description - resources.projects.resources.locations.resources.nodes.methods.performMaintenance.flatPath - resources.projects.resources.locations.resources.nodes.methods.performMaintenance.httpMethod - resources.projects.resources.locations.resources.nodes.methods.performMaintenance.id - resources.projects.resources.locations.resources.nodes.methods.performMaintenance.parameterOrder - resources.projects.resources.locations.resources.nodes.methods.performMaintenance.parameters.name.description - resources.projects.resources.locations.resources.nodes.methods.performMaintenance.parameters.name.location - resources.projects.resources.locations.resources.nodes.methods.performMaintenance.parameters.name.pattern - resources.projects.resources.locations.resources.nodes.methods.performMaintenance.parameters.name.required - resources.projects.resources.locations.resources.nodes.methods.performMaintenance.parameters.name.type - resources.projects.resources.locations.resources.nodes.methods.performMaintenance.path - resources.projects.resources.locations.resources.nodes.methods.performMaintenance.request.$ref - resources.projects.resources.locations.resources.nodes.methods.performMaintenance.response.$ref - resources.projects.resources.locations.resources.nodes.methods.performMaintenance.scopes - resources.projects.resources.locations.resources.queuedResources.methods.performMaintenanceQueuedResource.description - resources.projects.resources.locations.resources.queuedResources.methods.performMaintenanceQueuedResource.flatPath - resources.projects.resources.locations.resources.queuedResources.methods.performMaintenanceQueuedResource.httpMethod - resources.projects.resources.locations.resources.queuedResources.methods.performMaintenanceQueuedResource.id - resources.projects.resources.locations.resources.queuedResources.methods.performMaintenanceQueuedResource.parameterOrder - resources.projects.resources.locations.resources.queuedResources.methods.performMaintenanceQueuedResource.parameters.name.description - resources.projects.resources.locations.resources.queuedResources.methods.performMaintenanceQueuedResource.parameters.name.location - resources.projects.resources.locations.resources.queuedResources.methods.performMaintenanceQueuedResource.parameters.name.pattern - resources.projects.resources.locations.resources.queuedResources.methods.performMaintenanceQueuedResource.parameters.name.required - resources.projects.resources.locations.resources.queuedResources.methods.performMaintenanceQueuedResource.parameters.name.type - resources.projects.resources.locations.resources.queuedResources.methods.performMaintenanceQueuedResource.path - resources.projects.resources.locations.resources.queuedResources.methods.performMaintenanceQueuedResource.request.$ref - resources.projects.resources.locations.resources.queuedResources.methods.performMaintenanceQueuedResource.response.$ref - resources.projects.resources.locations.resources.queuedResources.methods.performMaintenanceQueuedResource.scopes - schemas.MultiNodeParams.properties.workloadType.description - schemas.MultiNodeParams.properties.workloadType.enum - schemas.MultiNodeParams.properties.workloadType.enumDescriptions - schemas.MultiNodeParams.properties.workloadType.type - schemas.Node.properties.upcomingMaintenance.$ref - schemas.Node.properties.upcomingMaintenance.description - schemas.Node.properties.upcomingMaintenance.readOnly - schemas.PerformMaintenanceQueuedResourceRequest.description - schemas.PerformMaintenanceQueuedResourceRequest.id - schemas.PerformMaintenanceQueuedResourceRequest.properties.nodeNames.description - schemas.PerformMaintenanceQueuedResourceRequest.properties.nodeNames.items.type - schemas.PerformMaintenanceQueuedResourceRequest.properties.nodeNames.type - schemas.PerformMaintenanceQueuedResourceRequest.type - schemas.PerformMaintenanceRequest.description - schemas.PerformMaintenanceRequest.id - schemas.PerformMaintenanceRequest.type - schemas.Reservation.properties.standard.description - schemas.Standard.description - schemas.Standard.properties.capacityUnits.description - schemas.Standard.properties.usage.description - schemas.UpcomingMaintenance.description - schemas.UpcomingMaintenance.id - schemas.UpcomingMaintenance.properties.canReschedule.description - schemas.UpcomingMaintenance.properties.canReschedule.type - schemas.UpcomingMaintenance.properties.latestWindowStartTime.description - schemas.UpcomingMaintenance.properties.latestWindowStartTime.type - schemas.UpcomingMaintenance.properties.maintenanceStatus.description - schemas.UpcomingMaintenance.properties.maintenanceStatus.enum - schemas.UpcomingMaintenance.properties.maintenanceStatus.enumDescriptions - schemas.UpcomingMaintenance.properties.maintenanceStatus.type - schemas.UpcomingMaintenance.properties.type.description - schemas.UpcomingMaintenance.properties.type.enum - schemas.UpcomingMaintenance.properties.type.enumDescriptions - schemas.UpcomingMaintenance.properties.type.type - schemas.UpcomingMaintenance.properties.windowEndTime.description - schemas.UpcomingMaintenance.properties.windowEndTime.type - schemas.UpcomingMaintenance.properties.windowStartTime.description - schemas.UpcomingMaintenance.properties.windowStartTime.type - schemas.UpcomingMaintenance.type - schemas.Usage.description The following keys were changed: - resources.projects.resources.locations.resources.operations.methods.cancel.description - resources.projects.resources.locations.resources.reservations.methods.list.parameters.pageSize.description - schemas.AcceleratorConfig.properties.type.enum - schemas.AcceleratorConfig.properties.type.enumDescriptions - schemas.Node.properties.networkConfig.description - schemas.Node.properties.networkConfigs.description - schemas.Node.properties.state.enum - schemas.Node.properties.state.enumDescriptions #### tpu:v1 The following keys were changed: - resources.projects.resources.locations.resources.operations.methods.cancel.description - schemas.Node.properties.state.enum - schemas.Node.properties.state.enumDescriptions #### tpu:v2 The following keys were added: - schemas.Node.properties.networkConfigs.description - schemas.Node.properties.networkConfigs.items.$ref - schemas.Node.properties.networkConfigs.type The following keys were changed: - resources.projects.resources.locations.resources.operations.methods.cancel.description - schemas.AcceleratorConfig.properties.type.enum - schemas.AcceleratorConfig.properties.type.enumDescriptions - schemas.MultisliceParams.properties.nodeCount.description - schemas.Node.properties.networkConfig.description - schemas.Node.properties.state.enum - schemas.Node.properties.state.enumDescriptions
1 parent 0e6f86a commit 5c0a386

File tree

8 files changed

+478
-36
lines changed

8 files changed

+478
-36
lines changed

‎discovery/tpu-v1.json

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -462,7 +462,7 @@
462462
"operations": {
463463
"methods": {
464464
"cancel": {
465-
"description": "Starts asynchronous cancellation on a long-running operation. The server makes a best effort to cancel the operation, but success is not guaranteed. If the server doesn't support this method, it returns `google.rpc.Code.UNIMPLEMENTED`. Clients can use Operations.GetOperation or other methods to check whether the cancellation succeeded or whether the operation completed despite cancellation. On successful cancellation, the operation is not deleted; instead, it becomes an operation with an Operation.error value with a google.rpc.Status.code of 1, corresponding to `Code.CANCELLED`.",
465+
"description": "Starts asynchronous cancellation on a long-running operation. The server makes a best effort to cancel the operation, but success is not guaranteed. If the server doesn't support this method, it returns `google.rpc.Code.UNIMPLEMENTED`. Clients can use Operations.GetOperation or other methods to check whether the cancellation succeeded or whether the operation completed despite cancellation. On successful cancellation, the operation is not deleted; instead, it becomes an operation with an Operation.error value with a google.rpc.Status.code of `1`, corresponding to `Code.CANCELLED`.",
466466
"flatPath": "v1/projects/{projectsId}/locations/{locationsId}/operations/{operationsId}:cancel",
467467
"httpMethod": "POST",
468468
"id": "tpu.projects.locations.operations.cancel",
@@ -659,7 +659,7 @@
659659
}
660660
}
661661
},
662-
"revision": "20230803",
662+
"revision": "20241126",
663663
"rootUrl": "https://tpu.googleapis.com/",
664664
"schemas": {
665665
"AcceleratorType": {
@@ -968,7 +968,8 @@
968968
"TERMINATED",
969969
"HIDING",
970970
"HIDDEN",
971-
"UNHIDING"
971+
"UNHIDING",
972+
"UNKNOWN"
972973
],
973974
"enumDescriptions": [
974975
"TPU node state is not known/set.",
@@ -985,7 +986,8 @@
985986
"TPU node has been terminated due to maintenance or has reached the end of its life cycle (for preemptible nodes).",
986987
"TPU node is currently hiding.",
987988
"TPU node has been hidden.",
988-
"TPU node is currently unhiding."
989+
"TPU node is currently unhiding.",
990+
"TPU node has unknown state after a failed repair."
989991
],
990992
"readOnly": true,
991993
"type": "string"

‎discovery/tpu-v1alpha1.json

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -472,7 +472,7 @@
472472
"operations": {
473473
"methods": {
474474
"cancel": {
475-
"description": "Starts asynchronous cancellation on a long-running operation. The server makes a best effort to cancel the operation, but success is not guaranteed. If the server doesn't support this method, it returns `google.rpc.Code.UNIMPLEMENTED`. Clients can use Operations.GetOperation or other methods to check whether the cancellation succeeded or whether the operation completed despite cancellation. On successful cancellation, the operation is not deleted; instead, it becomes an operation with an Operation.error value with a google.rpc.Status.code of 1, corresponding to `Code.CANCELLED`.",
475+
"description": "Starts asynchronous cancellation on a long-running operation. The server makes a best effort to cancel the operation, but success is not guaranteed. If the server doesn't support this method, it returns `google.rpc.Code.UNIMPLEMENTED`. Clients can use Operations.GetOperation or other methods to check whether the cancellation succeeded or whether the operation completed despite cancellation. On successful cancellation, the operation is not deleted; instead, it becomes an operation with an Operation.error value with a google.rpc.Status.code of `1`, corresponding to `Code.CANCELLED`.",
476476
"flatPath": "v1alpha1/projects/{projectsId}/locations/{locationsId}/operations/{operationsId}:cancel",
477477
"httpMethod": "POST",
478478
"id": "tpu.projects.locations.operations.cancel",
@@ -669,7 +669,7 @@
669669
}
670670
}
671671
},
672-
"revision": "20230803",
672+
"revision": "20241126",
673673
"rootUrl": "https://tpu.googleapis.com/",
674674
"schemas": {
675675
"AcceleratorType": {
@@ -978,7 +978,8 @@
978978
"TERMINATED",
979979
"HIDING",
980980
"HIDDEN",
981-
"UNHIDING"
981+
"UNHIDING",
982+
"UNKNOWN"
982983
],
983984
"enumDescriptions": [
984985
"TPU node state is not known/set.",
@@ -995,7 +996,8 @@
995996
"TPU node has been terminated due to maintenance or has reached the end of its life cycle (for preemptible nodes).",
996997
"TPU node is currently hiding.",
997998
"TPU node has been hidden.",
998-
"TPU node is currently unhiding."
999+
"TPU node is currently unhiding.",
1000+
"TPU node has unknown state after a failed repair."
9991001
],
10001002
"readOnly": true,
10011003
"type": "string"

‎discovery/tpu-v2.json

Lines changed: 19 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -524,7 +524,7 @@
524524
"operations": {
525525
"methods": {
526526
"cancel": {
527-
"description": "Starts asynchronous cancellation on a long-running operation. The server makes a best effort to cancel the operation, but success is not guaranteed. If the server doesn't support this method, it returns `google.rpc.Code.UNIMPLEMENTED`. Clients can use Operations.GetOperation or other methods to check whether the cancellation succeeded or whether the operation completed despite cancellation. On successful cancellation, the operation is not deleted; instead, it becomes an operation with an Operation.error value with a google.rpc.Status.code of 1, corresponding to `Code.CANCELLED`.",
527+
"description": "Starts asynchronous cancellation on a long-running operation. The server makes a best effort to cancel the operation, but success is not guaranteed. If the server doesn't support this method, it returns `google.rpc.Code.UNIMPLEMENTED`. Clients can use Operations.GetOperation or other methods to check whether the cancellation succeeded or whether the operation completed despite cancellation. On successful cancellation, the operation is not deleted; instead, it becomes an operation with an Operation.error value with a google.rpc.Status.code of `1`, corresponding to `Code.CANCELLED`.",
528528
"flatPath": "v2/projects/{projectsId}/locations/{locationsId}/operations/{operationsId}:cancel",
529529
"httpMethod": "POST",
530530
"id": "tpu.projects.locations.operations.cancel",
@@ -887,7 +887,7 @@
887887
}
888888
}
889889
},
890-
"revision": "20240613",
890+
"revision": "20250116",
891891
"rootUrl": "https://tpu.googleapis.com/",
892892
"schemas": {
893893
"AcceleratorConfig": {
@@ -906,15 +906,17 @@
906906
"V3",
907907
"V4",
908908
"V5LITE_POD",
909-
"V5P"
909+
"V5P",
910+
"V6E"
910911
],
911912
"enumDescriptions": [
912913
"Unspecified version.",
913914
"TPU v2.",
914915
"TPU v3.",
915916
"TPU v4.",
916917
"TPU v5lite pod.",
917-
"TPU v5p"
918+
"TPU v5p.",
919+
"TPU v6e."
918920
],
919921
"type": "string"
920922
}
@@ -1323,7 +1325,7 @@
13231325
"id": "MultisliceParams",
13241326
"properties": {
13251327
"nodeCount": {
1326-
"description": "Required. Number of nodes with this spec. The system will attempt to provison \"node_count\" nodes as part of the request. This needs to be > 1.",
1328+
"description": "Required. Number of nodes with this spec. The system will attempt to provision \"node_count\" nodes as part of the request. This needs to be > 1.",
13271329
"format": "int32",
13281330
"type": "integer"
13291331
},
@@ -1489,7 +1491,14 @@
14891491
},
14901492
"networkConfig": {
14911493
"$ref": "NetworkConfig",
1492-
"description": "Network configurations for the TPU node."
1494+
"description": "Network configurations for the TPU node. network_config and network_configs are mutually exclusive, you can only specify one of them. If both are specified, an error will be returned."
1495+
},
1496+
"networkConfigs": {
1497+
"description": "Optional. Repeated network configurations for the TPU node. This field is used to specify multiple networks configs for the TPU node. network_config and network_configs are mutually exclusive, you can only specify one of them. If both are specified, an error will be returned.",
1498+
"items": {
1499+
"$ref": "NetworkConfig"
1500+
},
1501+
"type": "array"
14931502
},
14941503
"networkEndpoints": {
14951504
"description": "Output only. The network endpoints where TPU workers can be accessed and sent work. It is recommended that runtime clients of the node reach out to the 0th entry in this map first.",
@@ -1537,7 +1546,8 @@
15371546
"TERMINATED",
15381547
"HIDING",
15391548
"HIDDEN",
1540-
"UNHIDING"
1549+
"UNHIDING",
1550+
"UNKNOWN"
15411551
],
15421552
"enumDescriptions": [
15431553
"TPU node state is not known/set.",
@@ -1554,7 +1564,8 @@
15541564
"TPU node has been terminated due to maintenance or has reached the end of its life cycle (for preemptible nodes).",
15551565
"TPU node is currently hiding.",
15561566
"TPU node has been hidden.",
1557-
"TPU node is currently unhiding."
1567+
"TPU node is currently unhiding.",
1568+
"TPU node has unknown state after a failed repair."
15581569
],
15591570
"readOnly": true,
15601571
"type": "string"

0 commit comments

Comments
 (0)