mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 11:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Suppress gRPC log spam on Cloud TPU.

Browse Source
This commit is contained in:
Skye Wanderman-Milne 2021-03-11 22:52:54 +00:00
parent 077793cd64
commit 5cb5056ea7
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
jax/_src/cloud_tpu_init.py
View File

@ -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([