package cloud_gvisor.tpu

Get desktop application:
View/edit binary Protocol Buffers messages

Used in: ControlResponse

(message has no fields)

Used in: ControlRequest

• ACTION_UNSPECIFIED = 0

• ACTION_LOCK = 1

• ACTION_CHECKPOINT = 2

• ACTION_RESTORE = 3

• ACTION_UNLOCK = 4

• ACTION_GETSTATE = 5

• optional ControlAction action = 1

• optional int32 timeout_secs = 2

• optional TpuTopology expected_topology = 3

  Used during ACTION_RESTORE. The runtime verifies this matches the physical hardware it is currently sitting on before proceeding.

• optional bool success = 1

• optional RuntimeState current_state = 2

• optional string error_message = 3

• optional TpuTopology active_topology = 4

• oneof telemetry

  • CheckpointTelemetry checkpoint_stats = 5

  • RestoreTelemetry restore_stats = 6

Used in: ControlResponse

(message has no fields)

Used in: ControlResponse

• STATE_UNSPECIFIED = 0

• STATE_RUNNING = 1

  Workload active

• STATE_LOCKED = 2

  PjRT API locked

• STATE_DETACHED = 3

  Safe to snapshot

• STATE_RESTORING = 4

  Hydrating TPU state

• STATE_FAULTED = 5

  Unrecoverable error

Prevents migration mismatches

Used in: ControlRequest, ControlResponse

• optional string chip_architecture = 1

  e.g., "v5e", "v5p"

• optional int32 num_chips = 2

  e.g., 4

• optional string mesh_shape = 3

  e.g., "2x2x1"

• optional string pjrt_version = 4

  Ensures the restored PjRT API matches the checkpointed API