From d52b6ddef422b7cb5d1e01ccf979658f5149e6b8 Mon Sep 17 00:00:00 2001
From: Peter Hawkins <phawkins@google.com>
Date: Tue, 21 May 2019 07:27:13 -0700
Subject: [PATCH] [XLA:Python] Enable asynchronous mode on GPU.
PiperOrigin-RevId: 249245334
---
tensorflow/compiler/xla/python/xla_client.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/tensorflow/compiler/xla/python/xla_client.py b/tensorflow/compiler/xla/python/xla_client.py
index 06681c55172..2cac98a422d 100644
--- a/tensorflow/compiler/xla/python/xla_client.py
+++ b/tensorflow/compiler/xla/python/xla_client.py
@@ -143,7 +143,7 @@ def _gpu_backend_factory():
config.memory_fraction = float(memory_fraction)
client = _xla.LocalClient.Get(
- platform='gpu', xla_platform_id='CUDA', asynchronous=False,
+ platform='gpu', xla_platform_id='CUDA', asynchronous=True,
allocator_config=config)
return LocalBackend(platform='gpu', client=client)