experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
STT-tensorflow/tensorflow/compiler/xla/python
History
Skye Wanderman-Milne 67ff36f392 [PJRT] Expose TPU runtime version via new PjRtClient::platform_version string. 
On Cloud TPU VMs, the returned platform version will look like:

libtpu version 0.0.1
Built on Mar 4 2021 15:25:57 (1614900357) cl/360760169

On all other platform, the platform version will be "<unknown>".

[XLA:Python] Add xla_client.Client.platform_version

PiperOrigin-RevId: 361215574
Change-Id: Ie65ce509caa550277972751cfb58027ad58c1d23
2021-03-05 13:45:22 -08:00
..
tpu_driver
[PJRT] Expose TPU runtime version via new PjRtClient::platform_version string.
2021-03-05 13:45:22 -08:00
__init__.py
…
absl_casters.h
[PJRT] Use absl::string_view to return non-owned strings from PJRT interfaces. NFC intended.
2021-01-25 07:17:02 -08:00
BUILD
[XLA:Python] Remove unnecessary NumPy API initialization.
2021-02-08 07:49:02 -08:00
custom_call_for_test.pyx
[XLA:Python] Add NextAfter to Python client.
2019-10-21 11:20:02 -07:00
dlpack.cc
[PJRT] Use ExternalReference as the return type of PjRtBuffer::ReleaseDeviceMemoryOwnership.
2021-02-01 19:07:22 -08:00
dlpack.h
[XLA:Python] Add support for DLPack buffer sharing without taking ownership.
2020-10-20 06:28:31 -07:00
jax_jit.cc
Deal with disabled-jit sooner.
2021-03-04 06:33:41 -08:00
jax_jit.h
Remove a callback to Python to get the value of some flag.
2021-03-02 05:44:12 -08:00
ops.cc
[XLA] Split QR decomposition along the same lines as LAPACK xGEQRf/xORGQR.
2021-02-23 14:13:25 -08:00
ops.h
[XLA:Python] Split bindings for XLA ops into a separate file. No functional changes.
2020-05-26 18:38:07 -07:00
outfeed_receiver_py.cc
[PJRT] Rename Device to PjRtDevice for consistency with the other PjRt classes.
2020-08-19 11:58:30 -07:00
outfeed_receiver_py.h
An implementation of a multithreaded runtime for receiving outfeed data
2020-05-27 05:52:21 -07:00
outfeed_receiver_test.cc
[PJRT] Rename PjRtDevice::host_id and PjRtClient::host_id to ...::task_id instead.
2021-01-26 17:32:20 -08:00
outfeed_receiver.cc
Allow PJRT infeed/outfeed methods to be non-const.
2021-02-02 22:00:28 -08:00
outfeed_receiver.h
[PJRT] Rename Device to PjRtDevice for consistency with the other PjRt classes.
2020-08-19 11:58:30 -07:00
pmap_lib.cc
Remove a callback to Python to get the value of some flag.
2021-03-02 05:44:12 -08:00
pmap_lib.h
Add some __eq__ functions to some objects that we can compare.
2021-02-22 15:48:52 -08:00
profiler.cc
Split out submodules from xla_extension.
2020-11-03 11:51:34 -08:00
profiler.h
Split out submodules from xla_extension.
2020-11-03 11:51:34 -08:00
py_buffer.cc
[XLA:Python] Expose buffer addresses in Python.
2021-03-02 16:57:49 -08:00
py_buffer.h
[PJRT] Remove on_host_shape from PjRtBuffer API.
2021-01-28 15:12:53 -08:00
py_client.cc
[PJRT] Change PjRtClient::BufferFromHostBuffer to take an on_done_with_host_buffer callback rather than a std::shared_ptr<void> that must be kept alive.
2021-01-27 07:49:33 -08:00
py_client.h
[PJRT] Expose TPU runtime version via new PjRtClient::platform_version string.
2021-03-05 13:45:22 -08:00
py_executable.cc
Document more explicitly the order of the arguments.
2021-02-23 14:56:13 -08:00
py_executable.h
Document more explicitly the order of the arguments.
2021-02-23 14:56:13 -08:00
py_traceback.cc
Split out submodules from xla_extension.
2020-11-03 11:51:34 -08:00
py_traceback.h
Split out submodules from xla_extension.
2020-11-03 11:51:34 -08:00
python_ref_manager.cc
[XLA:Python] Optimize traceback implementation.
2020-06-05 18:42:03 -07:00
python_ref_manager.h
[XLA:Python] Optimize traceback implementation.
2020-06-05 18:42:03 -07:00
pytree.cc
[XLA:Python] Clean up build dependencies. NFC.
2021-01-19 07:54:26 -08:00
pytree.h
Make it possible to stop PyTree flattening before reaching leaves.
2020-12-08 11:37:36 -08:00
traceback.cc
[XLA:Python] Implement on-device heap profiling support for JAX.
2020-06-12 11:18:13 -07:00
traceback.h
[XLA:Python] Implement on-device heap profiling support for JAX.
2020-06-12 11:18:13 -07:00
types.cc
[XLA:Python] Simplify code to convert literals to and from Python.
2021-01-07 20:07:22 -08:00
types.h
[XLA:Python] Clean up build dependencies. NFC.
2021-01-19 07:54:26 -08:00
xla_client_backend_independent_test.py
Speed up Shape creation by avoiding unnecessary validations.
2020-11-22 20:40:07 -08:00
xla_client_test.py
[PJRT] Expose TPU runtime version via new PjRtClient::platform_version string.
2021-03-05 13:45:22 -08:00
xla_client.py
Remove a callback to Python to get the value of some flag.
2021-03-02 05:44:12 -08:00
xla_compiler.cc
[XLA:Python] Remove unnecessary py::class_ registration for xla::LiteralSlice, and implicit conversions for literals.
2021-02-01 08:46:15 -08:00
xla_compiler.h
Split out submodules from xla_extension.
2020-11-03 11:51:34 -08:00
xla.cc
[PJRT] Expose TPU runtime version via new PjRtClient::platform_version string.
2021-03-05 13:45:22 -08:00