[GCP] TPU v4 fails to launch

[romilbhardwaj]

$ sky launch -c test task.yaml --cloud gcp --gpus tpu-v4-16

I 08-16 17:27:27 provisioner.py:65] Launching on GCP us-central2 (us-central2-b)
W 08-16 17:27:37 instance_utils.py:112] Got return code 'CREATION_FAILED' in us-central2-b: 'Cloud TPU received a bad request. the accelerator v4-128 was not found in zone us-central2-b [EID: 0x945ef875d10e8e13]'
D 08-16 17:27:37 provisioner.py:171] Failed to provision 'test' on GCP (us-central2-b).

Happens for other TPU slices (8, 16, 32 ...) too.

Version & Commit info:

• sky -c: 69838f5e94b3f45b6b77e0791576be5299da2b01

[Michaelvll]

Just a note, it seems v5 is working fine.