Skip to content

Commit aca4795

Browse files
m-strzelczykm-strzelczyk
authored
chore(tpu): Updating TPUv2 to TPUv5 (GoogleCloudPlatform#10038)
* chore(tpu): Updating TPUv2 to TPUv5 * Fixing bad replaces * Another fix.... * Fixing TpuVMIT
1 parent 1975c4c commit aca4795

10 files changed

+22
-22
lines changed

‎tpu/src/main/java/tpu/CreateQueuedResource.java‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -35,16 +35,16 @@ public static void main(String[] args)
3535
// The zone in which to create the TPU.
3636
// For more information about supported TPU types for specific zones,
3737
// see https://cloud.google.com/tpu/docs/regions-zones
38-
String zone = "us-central1-f";
38+
String zone = "us-central1-a";
3939
// The name for your TPU.
4040
String nodeName = "YOUR_NODE_ID";
4141
// The accelerator type that specifies the version and size of the Cloud TPU you want to create.
4242
// For more information about supported accelerator types for each TPU version,
4343
// see https://cloud.google.com/tpu/docs/system-architecture-tpu-vm#versions.
44-
String tpuType = "v2-8";
44+
String tpuType = "v5litepod-4";
4545
// Software version that specifies the version of the TPU runtime to install.
4646
// For more information see https://cloud.google.com/tpu/docs/runtimes
47-
String tpuSoftwareVersion = "tpu-vm-tf-2.14.1";
47+
String tpuSoftwareVersion = "v2-tpuv5-litepod";
4848
// The name for your Queued Resource.
4949
String queuedResourceId = "QUEUED_RESOURCE_ID";
5050

‎tpu/src/main/java/tpu/CreateQueuedResourceWithNetwork.java‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -43,10 +43,10 @@ public static void main(String[] args)
4343
// The accelerator type that specifies the version and size of the Cloud TPU you want to create.
4444
// For more information about supported accelerator types for each TPU version,
4545
// see https://cloud.google.com/tpu/docs/system-architecture-tpu-vm#versions.
46-
String tpuType = "v2-8";
46+
String tpuType = "v5litepod-4";
4747
// Software version that specifies the version of the TPU runtime to install.
4848
// For more information see https://cloud.google.com/tpu/docs/runtimes
49-
String tpuSoftwareVersion = "tpu-vm-tf-2.14.1";
49+
String tpuSoftwareVersion = "v2-tpuv5-litepod";
5050
// The name for your Queued Resource.
5151
String queuedResourceId = "QUEUED_RESOURCE_ID";
5252
// The name of the network you want the node to connect to.

‎tpu/src/main/java/tpu/CreateQueuedResourceWithStartupScript.java‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -41,10 +41,10 @@ public static void main(String[] args)
4141
// The accelerator type that specifies the version and size of the Cloud TPU you want to create.
4242
// For more information about supported accelerator types for each TPU version,
4343
// see https://cloud.google.com/tpu/docs/system-architecture-tpu-vm#versions.
44-
String tpuType = "v2-8";
44+
String tpuType = "v5litepod-4";
4545
// Software version that specifies the version of the TPU runtime to install.
4646
// For more information see https://cloud.google.com/tpu/docs/runtimes
47-
String tpuSoftwareVersion = "tpu-vm-tf-2.14.1";
47+
String tpuSoftwareVersion = "v2-tpuv5-litepod";
4848
// The name for your Queued Resource.
4949
String queuedResourceId = "QUEUED_RESOURCE_ID";
5050

‎tpu/src/main/java/tpu/CreateSpotQueuedResource.java‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -34,16 +34,16 @@ public static void main(String[] args)
3434
// The zone in which to create the TPU.
3535
// For more information about supported TPU types for specific zones,
3636
// see https://cloud.google.com/tpu/docs/regions-zones
37-
String zone = "us-central1-f";
37+
String zone = "us-central1-a";
3838
// The name for your TPU.
3939
String nodeName = "YOUR_TPU_NAME";
4040
// The accelerator type that specifies the version and size of the Cloud TPU you want to create.
4141
// For more information about supported accelerator types for each TPU version,
4242
// see https://cloud.google.com/tpu/docs/system-architecture-tpu-vm#versions.
43-
String tpuType = "v2-8";
43+
String tpuType = "v5litepod-4";
4444
// Software version that specifies the version of the TPU runtime to install.
4545
// For more information see https://cloud.google.com/tpu/docs/runtimes
46-
String tpuSoftwareVersion = "tpu-vm-tf-2.14.1";
46+
String tpuSoftwareVersion = "v2-tpuv5-litepod";
4747
// The name for your Queued Resource.
4848
String queuedResourceId = "QUEUED_RESOURCE_ID";
4949

‎tpu/src/main/java/tpu/CreateSpotTpuVm.java‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -33,16 +33,16 @@ public static void main(String[] args)
3333
// The zone in which to create the TPU.
3434
// For more information about supported TPU types for specific zones,
3535
// see https://cloud.google.com/tpu/docs/regions-zones
36-
String zone = "us-central1-f";
36+
String zone = "us-central1-a";
3737
// The name for your TPU.
3838
String nodeName = "YOUR_TPY_NAME";
3939
// The accelerator type that specifies the version and size of the Cloud TPU you want to create.
4040
// For more information about supported accelerator types for each TPU version,
4141
// see https://cloud.google.com/tpu/docs/system-architecture-tpu-vm#versions.
42-
String tpuType = "v2-8";
42+
String tpuType = "v5litepod-4";
4343
// Software version that specifies the version of the TPU runtime to install.
4444
// For more information see https://cloud.google.com/tpu/docs/runtimes
45-
String tpuSoftwareVersion = "tpu-vm-tf-2.14.1";
45+
String tpuSoftwareVersion = "v2-tpuv5-litepod";
4646

4747
createSpotTpuVm(projectId, zone, nodeName, tpuType, tpuSoftwareVersion);
4848
}

‎tpu/src/main/java/tpu/CreateTimeBoundQueuedResource.java‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@ public static void main(String[] args)
4444
String acceleratorType = "v2-8";
4545
// Software version that specifies the version of the TPU runtime to install.
4646
// For more information see https://cloud.google.com/tpu/docs/runtimes
47-
String runtimeVersion = "tpu-vm-tf-2.14.1";
47+
String runtimeVersion = "v2-tpuv5-litepod";
4848
// The name of your Queued Resource.
4949
String queuedResourceId = "YOUR_QUEUED_RESOURCE_ID";
5050

‎tpu/src/main/java/tpu/CreateTpuVm.java‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ public static void main(String[] args)
4646
String tpuType = "v2-8";
4747
// Software version that specifies the version of the TPU runtime to install.
4848
// For more information see https://cloud.google.com/tpu/docs/runtimes
49-
String tpuSoftwareVersion = "tpu-vm-tf-2.14.1";
49+
String tpuSoftwareVersion = "v2-tpuv5-litepod";
5050

5151
createTpuVm(projectId, zone, nodeName, tpuType, tpuSoftwareVersion);
5252
}

‎tpu/src/main/java/tpu/CreateTpuVmWithStartupScript.java‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -34,16 +34,16 @@ public static void main(String[] args)
3434
// The zone in which to create the TPU.
3535
// For more information about supported TPU types for specific zones,
3636
// see https://cloud.google.com/tpu/docs/regions-zones
37-
String zone = "us-central1-f";
37+
String zone = "us-central1-a";
3838
// The name for your TPU.
3939
String nodeName = "YOUR_TPU_NAME";
4040
// The accelerator type that specifies the version and size of the Cloud TPU you want to create.
4141
// For more information about supported accelerator types for each TPU version,
4242
// see https://cloud.google.com/tpu/docs/system-architecture-tpu-vm#versions.
43-
String acceleratorType = "v2-8";
43+
String acceleratorType = "v5litepod-4";
4444
// Software version that specifies the version of the TPU runtime to install.
4545
// For more information, see https://cloud.google.com/tpu/docs/runtimes
46-
String tpuSoftwareVersion = "tpu-vm-tf-2.14.1";
46+
String tpuSoftwareVersion = "v2-tpuv5-litepod";
4747

4848
createTpuVmWithStartupScript(projectId, zone, nodeName, acceleratorType, tpuSoftwareVersion);
4949
}

‎tpu/src/test/java/tpu/QueuedResourceIT.java‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -53,8 +53,8 @@ public class QueuedResourceIT {
5353
private static final String PROJECT_ID = "project-id";
5454
private static final String ZONE = "europe-west4-a";
5555
private static final String NODE_NAME = "test-tpu";
56-
private static final String TPU_TYPE = "v2-8";
57-
private static final String TPU_SOFTWARE_VERSION = "tpu-vm-tf-2.14.1";
56+
private static final String TPU_TYPE = "v5litepod-4";
57+
private static final String TPU_SOFTWARE_VERSION = "v2-tpuv5-litepod";
5858
private static final String QUEUED_RESOURCE_NAME = "queued-resource";
5959
private static final String NETWORK_NAME = "default";
6060

‎tpu/src/test/java/tpu/TpuVmIT.java‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -54,9 +54,9 @@ public class TpuVmIT {
5454
private static final String PROJECT_ID = "project-id";
5555
private static final String ZONE = "asia-east1-c";
5656
private static final String NODE_NAME = "test-tpu";
57-
private static final String TPU_TYPE = "v2-8";
57+
private static final String TPU_TYPE = "v5litepod-4";
5858
private static final AcceleratorConfig.Type ACCELERATOR_TYPE = AcceleratorConfig.Type.V2;
59-
private static final String TPU_SOFTWARE_VERSION = "tpu-vm-tf-2.14.1";
59+
private static final String TPU_SOFTWARE_VERSION = "v2-tpuv5-litepod";
6060
private static final String TOPOLOGY = "2x2";
6161

6262
@Test

0 commit comments

Comments
 (0)