Set async as default for TFRT runtime

PiperOrigin-RevId: 316456716 Change-Id: Ifdffb129c4609108285892a7ead5758e67802641
2020-06-15 07:00:00 -07:00 · 2020-06-15 07:00:00 -07:00 · eabae7b8e9
commit eabae7b8e9
parent af52bd27dc
1 changed files with 1 additions and 1 deletions
--- a/tensorflow/python/eager/context.py
+++ b/tensorflow/python/eager/context.py
@ -427,7 +427,7 @@ class Context(object):
      raise ValueError(
          "execution_mode should be None/SYNC/ASYNC. Got %s" % execution_mode)
    if execution_mode is None:
-      execution_mode = SYNC
+      execution_mode = ASYNC if is_tfrt_enabled() else SYNC
    self._default_is_async = execution_mode == ASYNC
    self._lazy_remote_inputs_copy = None
    self._use_tfrt = is_tfrt_enabled()