Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 9 additions & 0 deletions mmv1/products/tpuv2/Vm.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -60,6 +60,12 @@ examples:
external_providers:
- random
- time
- name: tpu_v2_vm_autocheckpoint
primary_resource_id: tpu
min_version: beta
vars:
vm_name: test-tpu-autochkpt
skip_vcr: true
parameters:
- name: zone
type: String
Expand Down Expand Up @@ -448,3 +454,6 @@ properties:
A string used to uniquely distinguish a worker within a TPU node.
output: true
min_version: beta
- name: autocheckpointEnabled
type: Boolean
description: Whether Autocheckpoint is enabled.
13 changes: 13 additions & 0 deletions mmv1/templates/terraform/examples/tpu_v2_vm_autocheckpoint.tf.tmpl
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
data "google_tpu_v2_runtime_versions" "available" {
provider = google-beta
}

resource "google_tpu_v2_vm" "{{$.PrimaryResourceId}}" {
provider = google-beta

name = "{{index $.Vars "vm_name"}}"
zone = "us-central1-c"

runtime_version = "tpu-vm-tf-2.13.0"
autocheckpoint_enabled = true
}
Loading