Merge pull request #4 from bcgazen:bcg_main

PiperOrigin-RevId: 595495136
google-research · Jan 3, 2024 · 809b8fb · 809b8fb
2 parents 4b77394 + 2329bbb
commit 809b8fb
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/docs/cloud_tpu.md b/docs/cloud_tpu.md
@@ -43,7 +43,7 @@ machine running the browser to Jupyter.
     TPU_FLAGS=--xla_jf_conv_full_precision=true
 
     gcloud compute tpus tpu-vm create $TPU --project=$PROJECT --zone=$ZONE \
-    --accelerator-type=$TYPE --version=tpu-vm-tf-2.11.0-pod \
+    --accelerator-type=$TYPE --version=tpu-vm-tf-2.12.0-pod \
     --metadata="tensorflow-env-vars=-e LIBTPU_INIT_ARGS=$TPU_FLAGS"
     ```
 
@@ -162,7 +162,7 @@ pod. For example, to set xla_jf_conv_full_precision to true:
 
 ```shell
 gcloud compute tpus tpu-vm create $TPU --project=$PROJECT --zone=$ZONE \
-  --accelerator-type=v3-32 --version=tpu-vm-tf-2.11.0-pod \
+  --accelerator-type=v3-32 --version=tpu-vm-tf-2.12.0-pod \
   --metadata="tensorflow-env-vars=-e LIBTPU_INIT_ARGS=--xla_jf_conv_full_precision=true"
 ```
 

diff --git a/swirl_lm/example/fire/README.md b/swirl_lm/example/fire/README.md
@@ -9,7 +9,7 @@ This example simulates the Tubbs Fire.
    ```sh
    gcloud compute tpus tpu-vm create $TPU --project=$PROJECT \
       --zone=$ZONE \
-      --accelerator-type=$TYPE --version=tpu-vm-tf-2.11.0-pod \
+      --accelerator-type=$TYPE --version=tpu-vm-tf-2.12.0-pod \
       --metadata="tensorflow-env-vars=-e LIBTPU_INIT_ARGS=$TPU_FLAGS"
    ```