From 5cb5056ea7718815c44b170e14bb04dfdec7e064 Mon Sep 17 00:00:00 2001 From: Skye Wanderman-Milne Date: Thu, 11 Mar 2021 22:52:54 +0000 Subject: [PATCH] Suppress gRPC log spam on Cloud TPU. --- jax/_src/cloud_tpu_init.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/jax/_src/cloud_tpu_init.py b/jax/_src/cloud_tpu_init.py index 9f79a2017..59d083a03 100644 --- a/jax/_src/cloud_tpu_init.py +++ b/jax/_src/cloud_tpu_init.py @@ -31,6 +31,8 @@ def cloud_tpu_init(): if not _running_in_cloud_tpu_vm(): return + os.environ.setdefault('GRPC_VERBOSITY', 'ERROR') + # If the user has set any topology-related env vars, don't set any # automatically. if any([