diff options
author | 2018-04-23 15:30:12 -0700 | |
---|---|---|
committer | 2018-04-23 15:32:44 -0700 | |
commit | 01141932a9cdcd871310db141a66a47410c48ac0 (patch) | |
tree | fbd0243a55d2b7ce1d13ac9a0ab3ffc9a7e0e4f7 /tensorflow/compiler/tests/BUILD | |
parent | 9e1d93d28fe30171de3f6838028eeadb44b0d6fd (diff) |
Support executing ops eagerly through XLA
The ony real change is to add GpuDeviceInfo to XlaDevice.
It is used by eager runtime to retrieve default device context.
PiperOrigin-RevId: 193995586
Diffstat (limited to 'tensorflow/compiler/tests/BUILD')
-rw-r--r-- | tensorflow/compiler/tests/BUILD | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/tensorflow/compiler/tests/BUILD b/tensorflow/compiler/tests/BUILD index 46b86c53aa..ac2441cea0 100644 --- a/tensorflow/compiler/tests/BUILD +++ b/tensorflow/compiler/tests/BUILD @@ -309,6 +309,25 @@ tf_xla_py_test( ) tf_xla_py_test( + name = "eager_test", + size = "small", + srcs = ["eager_test.py"], + disabled_backends = [ + # TODO(b/78199195) Support XLA CPU devices in eager runtime + "cpu", + "cpu_ondemand", + # TODO(b/78468222) Enable GPU backend + "gpu", + ], + deps = [ + ":xla_test", + "//tensorflow/python:array_ops", + "//tensorflow/python:framework_for_generated_wrappers", + "//tensorflow/python:platform_test", + ], +) + +tf_xla_py_test( name = "fft_test", size = "medium", srcs = ["fft_test.py"], |