Skip to content

Commit

Permalink
feat(tpu): update the API
Browse files Browse the repository at this point in the history
#### tpu:v1alpha1
The following keys were added:
- schemas.Node.properties.symptoms.description
- schemas.Node.properties.symptoms.items.$ref
- schemas.Node.properties.symptoms.readOnly
- schemas.Node.properties.symptoms.type
- schemas.Symptom.description
- schemas.Symptom.id
- schemas.Symptom.properties.createTime.description
- schemas.Symptom.properties.createTime.format
- schemas.Symptom.properties.createTime.type
- schemas.Symptom.properties.details.description
- schemas.Symptom.properties.details.type
- schemas.Symptom.properties.symptomType.description
- schemas.Symptom.properties.symptomType.enum
- schemas.Symptom.properties.symptomType.enumDescriptions
- schemas.Symptom.properties.symptomType.type
- schemas.Symptom.properties.workerId.description
- schemas.Symptom.properties.workerId.type
- schemas.Symptom.type

#### tpu:v1
The following keys were added:
- schemas.Node.properties.symptoms.description
- schemas.Node.properties.symptoms.items.$ref
- schemas.Node.properties.symptoms.readOnly
- schemas.Node.properties.symptoms.type
- schemas.Symptom.description
- schemas.Symptom.id
- schemas.Symptom.properties.createTime.description
- schemas.Symptom.properties.createTime.format
- schemas.Symptom.properties.createTime.type
- schemas.Symptom.properties.details.description
- schemas.Symptom.properties.details.type
- schemas.Symptom.properties.symptomType.description
- schemas.Symptom.properties.symptomType.enum
- schemas.Symptom.properties.symptomType.enumDescriptions
- schemas.Symptom.properties.symptomType.type
- schemas.Symptom.properties.workerId.description
- schemas.Symptom.properties.workerId.type
- schemas.Symptom.type
  • Loading branch information
JustinBeckwith committed Oct 5, 2020
1 parent 0cde769 commit 897a43d
Show file tree
Hide file tree
Showing 4 changed files with 152 additions and 2 deletions.
50 changes: 49 additions & 1 deletion discovery/tpu-v1.json
Original file line number Diff line number Diff line change
Expand Up @@ -659,7 +659,7 @@
}
}
},
"revision": "20200814",
"revision": "20200909",
"rootUrl": "https://tpu.googleapis.com/",
"schemas": {
"AcceleratorType": {
Expand Down Expand Up @@ -964,6 +964,14 @@
],
"type": "string"
},
"symptoms": {
"description": "Output only. The Symptoms that have occurred to the TPU Node.",
"items": {
"$ref": "Symptom"
},
"readOnly": true,
"type": "array"
},
"tensorflowVersion": {
"description": "The version of Tensorflow running in the Node. Required.",
"type": "string"
Expand Down Expand Up @@ -1112,6 +1120,46 @@
"properties": {},
"type": "object"
},
"Symptom": {
"description": "A Symptom instance.",
"id": "Symptom",
"properties": {
"createTime": {
"description": "Timestamp when the Symptom is created.",
"format": "google-datetime",
"type": "string"
},
"details": {
"description": "Detailed information of the current Symptom.",
"type": "string"
},
"symptomType": {
"description": "Type of the Symptom.",
"enum": [
"SYMPTOM_TYPE_UNSPECIFIED",
"LOW_MEMORY",
"OUT_OF_MEMORY",
"EXECUTE_TIMED_OUT",
"MESH_BUILD_FAIL",
"HBM_OUT_OF_MEMORY"
],
"enumDescriptions": [
"Unspecified symptom.",
"TPU VM memory is low.",
"TPU runtime is out of memory.",
"TPU runtime execution has timed out.",
"TPU runtime fails to construct a mesh that recognizes each TPU device's neighbors.",
"TPU HBM is out of memory."
],
"type": "string"
},
"workerId": {
"description": "A string used to uniquely distinguish a worker within a TPU node.",
"type": "string"
}
},
"type": "object"
},
"TensorFlowVersion": {
"description": "A tensorflow version that a Node can be configured with.",
"id": "TensorFlowVersion",
Expand Down
50 changes: 49 additions & 1 deletion discovery/tpu-v1alpha1.json
Original file line number Diff line number Diff line change
Expand Up @@ -659,7 +659,7 @@
}
}
},
"revision": "20200814",
"revision": "20200909",
"rootUrl": "https://tpu.googleapis.com/",
"schemas": {
"AcceleratorType": {
Expand Down Expand Up @@ -964,6 +964,14 @@
],
"type": "string"
},
"symptoms": {
"description": "Output only. The Symptoms that have occurred to the TPU Node.",
"items": {
"$ref": "Symptom"
},
"readOnly": true,
"type": "array"
},
"tensorflowVersion": {
"description": "The version of Tensorflow running in the Node. Required.",
"type": "string"
Expand Down Expand Up @@ -1112,6 +1120,46 @@
"properties": {},
"type": "object"
},
"Symptom": {
"description": "A Symptom instance.",
"id": "Symptom",
"properties": {
"createTime": {
"description": "Timestamp when the Symptom is created.",
"format": "google-datetime",
"type": "string"
},
"details": {
"description": "Detailed information of the current Symptom.",
"type": "string"
},
"symptomType": {
"description": "Type of the Symptom.",
"enum": [
"SYMPTOM_TYPE_UNSPECIFIED",
"LOW_MEMORY",
"OUT_OF_MEMORY",
"EXECUTE_TIMED_OUT",
"MESH_BUILD_FAIL",
"HBM_OUT_OF_MEMORY"
],
"enumDescriptions": [
"Unspecified symptom.",
"TPU VM memory is low.",
"TPU runtime is out of memory.",
"TPU runtime execution has timed out.",
"TPU runtime fails to construct a mesh that recognizes each TPU device's neighbors.",
"TPU HBM is out of memory."
],
"type": "string"
},
"workerId": {
"description": "A string used to uniquely distinguish a worker within a TPU node.",
"type": "string"
}
},
"type": "object"
},
"TensorFlowVersion": {
"description": "A tensorflow version that a Node can be configured with.",
"id": "TensorFlowVersion",
Expand Down
27 changes: 27 additions & 0 deletions src/apis/tpu/v1.ts
Original file line number Diff line number Diff line change
Expand Up @@ -320,6 +320,10 @@ export namespace tpu_v1 {
* Output only. The current state for the TPU Node.
*/
state?: string | null;
/**
* Output only. The Symptoms that have occurred to the TPU Node.
*/
symptoms?: Schema$Symptom[];
/**
* The version of Tensorflow running in the Node. Required.
*/
Expand Down Expand Up @@ -434,6 +438,27 @@ export namespace tpu_v1 {
* Request for StopNode.
*/
export interface Schema$StopNodeRequest {}
/**
* A Symptom instance.
*/
export interface Schema$Symptom {
/**
* Timestamp when the Symptom is created.
*/
createTime?: string | null;
/**
* Detailed information of the current Symptom.
*/
details?: string | null;
/**
* Type of the Symptom.
*/
symptomType?: string | null;
/**
* A string used to uniquely distinguish a worker within a TPU node.
*/
workerId?: string | null;
}
/**
* A tensorflow version that a Node can be configured with.
*/
Expand Down Expand Up @@ -1162,6 +1187,7 @@ export namespace tpu_v1 {
* // "schedulingConfig": {},
* // "serviceAccount": "my_serviceAccount",
* // "state": "my_state",
* // "symptoms": [],
* // "tensorflowVersion": "my_tensorflowVersion",
* // "useServiceNetworking": false
* // }
Expand Down Expand Up @@ -1455,6 +1481,7 @@ export namespace tpu_v1 {
* // "schedulingConfig": {},
* // "serviceAccount": "my_serviceAccount",
* // "state": "my_state",
* // "symptoms": [],
* // "tensorflowVersion": "my_tensorflowVersion",
* // "useServiceNetworking": false
* // }
Expand Down
27 changes: 27 additions & 0 deletions src/apis/tpu/v1alpha1.ts
Original file line number Diff line number Diff line change
Expand Up @@ -320,6 +320,10 @@ export namespace tpu_v1alpha1 {
* Output only. The current state for the TPU Node.
*/
state?: string | null;
/**
* Output only. The Symptoms that have occurred to the TPU Node.
*/
symptoms?: Schema$Symptom[];
/**
* The version of Tensorflow running in the Node. Required.
*/
Expand Down Expand Up @@ -434,6 +438,27 @@ export namespace tpu_v1alpha1 {
* Request for StopNode.
*/
export interface Schema$StopNodeRequest {}
/**
* A Symptom instance.
*/
export interface Schema$Symptom {
/**
* Timestamp when the Symptom is created.
*/
createTime?: string | null;
/**
* Detailed information of the current Symptom.
*/
details?: string | null;
/**
* Type of the Symptom.
*/
symptomType?: string | null;
/**
* A string used to uniquely distinguish a worker within a TPU node.
*/
workerId?: string | null;
}
/**
* A tensorflow version that a Node can be configured with.
*/
Expand Down Expand Up @@ -1162,6 +1187,7 @@ export namespace tpu_v1alpha1 {
* // "schedulingConfig": {},
* // "serviceAccount": "my_serviceAccount",
* // "state": "my_state",
* // "symptoms": [],
* // "tensorflowVersion": "my_tensorflowVersion",
* // "useServiceNetworking": false
* // }
Expand Down Expand Up @@ -1455,6 +1481,7 @@ export namespace tpu_v1alpha1 {
* // "schedulingConfig": {},
* // "serviceAccount": "my_serviceAccount",
* // "state": "my_state",
* // "symptoms": [],
* // "tensorflowVersion": "my_tensorflowVersion",
* // "useServiceNetworking": false
* // }
Expand Down

0 comments on commit 897a43d

Please sign in to comment.