Skip to content
This repository was archived by the owner on Feb 3, 2025. It is now read-only.

Commit 1458d08

Browse files
author
DEKHTIARJonathan
committed
[Benchmarking Py] Absl Logging Cleanup
1 parent 1c3d83f commit 1458d08

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

tftrt/benchmarking-python/benchmark_runner.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,8 @@
1919
import scipy.stats
2020
import tensorflow as tf
2121

22+
from absl import logging as absl_logging
23+
2224
from tensorflow.python.compiler.tensorrt import trt_convert as trt
2325
from tensorflow.python.framework.errors_impl import OutOfRangeError
2426
from tensorflow.python.platform import tf_logging
@@ -101,6 +103,9 @@ def __init__(self, args):
101103
None
102104
)
103105

106+
# Set ABSL verbosity to Err level
107+
absl_logging.set_verbosity(absl_logging.ERROR)
108+
104109
# TensorFlow can execute operations synchronously or asynchronously.
105110
# If asynchronous execution is enabled, operations may return
106111
# "non-ready" handles.

0 commit comments

Comments
 (0)