experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
STT-tensorflow/tensorflow/compiler/xla/python/tpu_driver
History
Peter Hawkins 7df1f4ffc3 [XLA:Python] Add support for collecting Python tracebacks. 
Adds a new xla_client.Traceback API that can collect Python tracebacks cheaply (~2us). This is several orders of magnitude cheaper than using the Python `inspect.stack` API.

Add a facility to attach a traceback to every buffer and executable object describing its creation context. To avoid paying a runtime cost when not debugging, tracebacks collection is optional and disabled by default.

PiperOrigin-RevId: 314793876
Change-Id: Ie5f509364065739c1da4d1a8a729c4c6f56e2d03
2020-06-04 13:39:07 -07:00
..
client
[XLA:Python] Add support for collecting Python tracebacks.
2020-06-04 13:39:07 -07:00
platform/external
Refactoring some dependencies within TPU Driver
2019-12-19 15:03:18 -08:00
BUILD
Use tf_grpc_dependency in tensorflow/compiler/xla
2020-04-02 22:42:59 -07:00
direct_tpu_driver.cc
Add dependency for the direct TPU driver back into the tpu_driver target
2020-02-18 13:29:57 -08:00
event_id.h
TpuDriver: Improve handling of restarted clients/servers.
2019-11-15 14:04:48 -08:00
grpc_tpu_driver.cc
This CL optimizes C++11 range-based for loops where the variable is copied in each iteration but it would suffice to obtain it by const reference. This is only applied to loop variables of types that are expensive to copy which means they are not trivially copyable or have a non-trivial copy constructor or destructor.
2020-04-03 11:53:09 -07:00
grpc_tpu_driver.h
Don't use grpc_impl:: as it is not API
2019-12-02 09:07:16 -08:00
README.md
…
recording_tpu_driver.cc
This CL optimizes C++11 range-based for loops where the variable is copied in each iteration but it would suffice to obtain it by const reference. This is only applied to loop variables of types that are expensive to copy which means they are not trivially copyable or have a non-trivial copy constructor or destructor.
2020-04-03 11:53:09 -07:00
tpu_driver.cc
Add shape-based allocation and linearizer to external TPU driver
2020-01-09 11:43:21 -08:00
tpu_driver.h
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
tpu_driver.proto
Enable TPU POD for JAX/1VM by creating devices and local_devices topology.
2020-01-14 17:04:41 -08:00
tpu_service.proto
Implement GRPC TPU driver reset.
2019-11-20 17:49:06 -08:00

README.md

TPU Driver API

This repository contains the TPU driver API and network (gRPC) transport implementation for high-performance access to TPU hardware.

Building

Bazel is used to build the driver library and tests. Remote tests will require access to a Cloud TPU.

Fetching Bazel

Download the latest copy of Bazel from https://github.com/bazelbuild/bazel/releases.

Building

bazel build ...

Testing

bazel test ...