SrSSKJr SSKJr SSKJr SSKJr SSKJr SrSr S r S r S r S r S rSrSrSSjrSrSrSSjrSrSrSrSrSrSrSrSrSSjrSrg) z&Flag Utilities for cloud tpu commands.)absolute_import)division)unicode_literals) arg_parsers)basec,[R"SSS9$)Ntpu_idzName of the TPU.helprArgument4lib/googlecloudsdk/command_lib/compute/tpus/flags.py GetTPUNameArgrs   rc,[R"SSS9$)Nz --descriptionz(Specifies a text description of the TPU.r r rrrGetDescriptionFlagrs  5 77rc4[R"SSSSSS9$)z3Set argument for choosing the TPU Accelerator type.z--accelerator-typezv2-8c"UR5$N)lower)xs r(GetAcceleratorTypeFlag..*s QWWYrFz` TPU accelerator type for the TPU. If not specified, this defaults to `v2-8`. )defaulttyperequiredr r rrrGetAcceleratorTypeFlagr%s$     rc.[R"SSSS9$)z6Set argument for choosing the TPU Tensor Flow version.z --versionTz TensorFlow version for the TPU, such as `1.6`. For a list of available TensorFlow versions please see https://www.tensorflow.org/versions/. rr r rrrGetVersionFlagr!2s    rc.[R"SSSS9$)zSet Cidr Range for Cloud TPU.z--rangeFa CIDR Range for the TPU. The IP range that the TPU will select an IP address from. Must be in CIDR notation and a `/29` range, for example `192.168.0.0/29`. Errors will occur if the CIDR range has already been used for a currently existing TPU, the CIDR range conflicts with any networks in the user's provided network, or the provided network is peered with another network that is using that CIDR range. r r rrr GetRangeFlagr#=s    rc(URSSSSSS9$)Nz --preemptibleF store_truez Create a preemptible Cloud TPU, instead of a normal (non-preemptible) Cloud TPU. A preemptible Cloud TPU costs less per hour, but the Cloud TPU service can stop/terminate the node at any time. ractionrr  add_argumentparsers rAddPreemptibleFlagr,Ms+       rc"URSSS9$)Nexecution_group_namez* The execution group name to delete. r r(r*s r AddTpuNameArgr/Zs"    .  rc"URSSS9$)Nz--namezM Override the name to use for VMs and TPUs (defaults to your username). r r(r*s rAddTpuNameOverrideArgr1bs#    Q  rc(URSSSSSS9$)Nz--preemptible-vmFr%z Create a preemptible Compute Engine VM, instead of a normal (non-preemptible) VM. A preemptible VM costs less per hour, but the Compute Engine service can terminate the instance at any time. r&r(r*s rAddPreemptibleVmFlagr3js+       rNc:SnURSSU=(d US9$)Nz Set the version of TensorFlow to use when creating the Compute Engine VM and the Cloud TPU. (It defaults to auto-selecting the latest stable release.) z --tf-versionFr r(r+help_text_override help_texts rAddTfVersionFlagr8ws1 )     *  rcSn[X5 g)Nz Set the version of TensorFlow to the version originally set when creating the suspended Cloud TPU and Compute Engine VM . (It defaults to auto-selecting the latest stable release.) )r8r+r6s rAddTfVersionFlagForResumer;s 6.rc(URSSSSSS9$)Nz --vm-onlyr%Fzh Do not allocate a TPU, only allocate a VM (useful if you're not ready to run on a TPU yet). r'rrr r(r*s r AddVmOnlyFlagr>s+       rc>SnURSSSSU=(d US9$)Nz Do not allocate a VM, only allocate a TPU. To be used after the command has been run with a --vm-only flag and the user is ready to run on a TPU. Ensure that the name matches the name passed in when creating with the --vm-only flag. z --tpu-onlyr%Fr=r(r5s rAddTpuOnlyFlagr@s7 )      *  ,,rcSn[X5$)Nz- Do not delete VM, only delete the TPU. )r@r:s rAddTpuOnlyFlagForDeleterBs  33rc(URSSSSSS9$)Nz--use-dl-imagesr%Fa Use Deep Learning VM Images (see docs - https://cloud.google.com/deep-learning-vm/) instead of TPU-specific machine images. Defaults to TPU-specific images. This value is set to true automatically if the --use-with-notebook flag is set to true. r=r(r*s rAddDeepLearningImagesFlagrDs+       rc(URSSSSSS9$)Nz --dry-runFr%zF Do not make changes; print only what would have happened. r&r(r*s r AddDryRunFlagrFs+       rc&URSSSSS9$)Nz--forward-ports store_falseFz Automatically forward useful ports from the Compute Engine VM to your local machine. The ports forwarded are: 6006 (tensorboard), 8888 (jupyter notebooks), 8470 (TPU port), 8466 (TPU profiler port). )r'rr r(r*s rAddPortForwardingFlagrIs(       rc"URSSS9$)Nz --gce-imagez Override the automatically chosen Compute Engine Image. Use this flag when you're using your own custom images instead of the provided ones with TensorFlow pre-installed. r r(r*s rAddGceImageFlagrKs"       rc PURSS[R"SSS/S9SS9$) Nz --disk-size250GB20GB2000GBGB) lower_bound upper_boundsuggested_binary_size_scalesz Configures the root volume size of your Compute Engine VM (in GB). The minimum size is 20GB and the maximum is 2000GB. Specified value must be an integer multiple of Gigabytes. )rrr )r)r BinarySizer*s rAddDiskSizeFlagrUs>     ! !(,v /    rc$URSSSS9$)Nz--machine-typez n1-standard-1z Specifies the machine type used for the Compute Engine VM. To get a list of available machine types, run 'gcloud compute machine-types list'. If unspecified, the default type is n1-standard-1. rr r(r*s rAddMachineTypeArgsrXs%       rc(URSSSSSS9$)Nz--use-with-notebookr%Fz Allow Compute Engine VM to be recognized by Cloud AI Notebooks. This automatically sets the content of the flag --use-dl-images flag to be true. r=r(r*s rAddUseWithNotebookrZs+       rc:SnURSSU=(d US9$)Nz Specifies the network the Cloud TPU and associated VM should be created in. If unspecified, the network "default" is picked. z --networkrrWr(r5s rAddNetworkArgsr\s1)     *  ,,rcSn[X5$)Nz Set to the network that was originally used creating the suspended Cloud TPU and Compute Engine VM. (It defaults to using the 'default' network.) )r\r:s rAddNetworkArgsForResumer^ s  33rr)__doc__ __future__rrrgooglecloudsdk.callioperrrrrr!r#r,r/r1r3r8r;r>r@rBrDrFrIrKrUrXrZr\r^rrrrbs-&'/( 7         /  ,4          ,4r