feat: tpu_vm_start, tpu_vm_stop

Author: Joanna Grycz

compute/test/tpu.test.js

@@ -66,6 +66,22 @@ describe('Compute tpu', async () => {
     assert(Array.isArray(response));
   });
 
+  it('should stop tpu node', () => {
+    const response = execSync(`node ./tpu/vmStop.js ${nodeName} ${zone}`, {
+      cwd,
+    });
+
+    assert(response.includes(`Node: ${nodeName} stopped.`));
+  });
+
+  it('should start tpu node', () => {
+    const response = execSync(`node ./tpu/vmStart.js ${nodeName} ${zone}`, {
+      cwd,
+    });
+
+    assert(response.includes(`Node: ${nodeName} started.`));
+  });
+
   it('should delete tpu node', () => {
     const response = execSync(`node ./tpu/vmDelete.js ${nodeName} ${zone}`, {
       cwd,

compute/tpu/vmCreate.js

@@ -39,7 +39,7 @@ async function main(nodeName, zone, tpuType, tpuSoftwareVersion) {
   const region = 'europe-west4';
 
   // The name for your TPU.
-  // nodeName = 'node-name-2';
+  // nodeName = 'node-name-1';
 
   // The zone in which to create the TPU.
   // For more information about supported TPU types for specific zones,

compute/tpu/vmDelete.js

@@ -34,7 +34,7 @@ async function main(nodeName, zone) {
   // nodeName = 'node-name-1';
 
   // The zone, where the TPU is created.
-  // zone = 'us-central1-a';
+  // zone = 'europe-west4-a';
 
   async function callDeleteTpuVM() {
     const request = {

compute/tpu/vmGet.js

@@ -34,7 +34,7 @@ async function main(nodeName, zone) {
   // nodeName = 'node-name-1';
 
   // The zone, where the TPU is created.
-  // zone = 'us-central1-a';
+  // zone = 'europe-west4-a';
 
   async function callGetTpuVM() {
     const request = {

compute/tpu/vmList.js

@@ -31,7 +31,7 @@ async function main(zone) {
   const projectId = await tpuClient.getProjectId();
 
   // The zone from which the TPUs are retrived.
-  //   zone = 'us-central1-a';
+  //   zone = 'europe-west4-a';
 
   async function callTpuVMList() {
     const request = {

compute/tpu/vmStart.js

@@ -0,0 +1,58 @@
+/*
+ * Copyright 2024 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+'use strict';
+
+async function main(nodeName, zone) {
+  // [START tpu_vm_start]
+  // Import TpuClient
+  const {TpuClient} = require('@google-cloud/tpu').v2;
+
+  // Instantiate a tpuClient
+  const tpuClient = new TpuClient();
+
+  /**
+   * TODO(developer): Update/uncomment these variables before running the sample.
+   */
+  // Project ID or project number of the Google Cloud project you want to start a node.
+  const projectId = await tpuClient.getProjectId();
+
+  // The name of TPU to start.
+  //   nodeName = 'node-name-1';
+
+  // The zone, where the TPU is created.
+  //   zone = 'europe-west4-a';
+
+  async function callStartTpuVM() {
+    const request = {
+      name: `projects/${projectId}/locations/${zone}/nodes/${nodeName}`,
+    };
+
+    const [operation] = await tpuClient.startNode(request);
+    // Wait for the operation to complete.
+    await operation.promise();
+
+    console.log(`Node: ${nodeName} started.`);
+  }
+
+  await callStartTpuVM();
+  // [END tpu_vm_start]
+}
+
+main(...process.argv.slice(2)).catch(err => {
+  console.error(err);
+  process.exitCode = 1;
+});

compute/tpu/vmStop.js

@@ -0,0 +1,58 @@
+/*
+ * Copyright 2024 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+'use strict';
+
+async function main(nodeName, zone) {
+  // [START tpu_vm_stop]
+  // Import TpuClient
+  const {TpuClient} = require('@google-cloud/tpu').v2;
+
+  // Instantiate a tpuClient
+  const tpuClient = new TpuClient();
+
+  /**
+   * TODO(developer): Update/uncomment these variables before running the sample.
+   */
+  // Project ID or project number of the Google Cloud project you want to stop a node.
+  const projectId = await tpuClient.getProjectId();
+
+  // The name of TPU to stop.
+  //   nodeName = 'node-name-1';
+
+  // The zone, where the TPU is created.
+  //   zone = 'europe-west4-a';
+
+  async function callStopTpuVM() {
+    const request = {
+      name: `projects/${projectId}/locations/${zone}/nodes/${nodeName}`,
+    };
+
+    const [operation] = await tpuClient.stopNode(request);
+    // Wait for the operation to complete.
+    await operation.promise();
+
+    console.log(`Node: ${nodeName} stopped.`);
+  }
+
+  await callStopTpuVM();
+  // [END tpu_vm_stop]
+}
+
+main(...process.argv.slice(2)).catch(err => {
+  console.error(err);
+  process.exitCode = 1;
+});