experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
STT-tensorflow/tensorflow/stream_executor/host
History
Peter Hawkins 077b553fda [XLA:Python] Specify a 2MiB stack size for host stream threads. 
[StreamExecutor] Allow HostExecutor users to control the stack sizes of threads used for HostStream via.

Also include non_portable_tags in the keys used when creating an Executor. There seems to be no good reason that it is omitted.

Will fix https://github.com/google/jax/issues/432 when included in a jaxlib release.

PiperOrigin-RevId: 309472318
Change-Id: Ia2535616047390d6bf6f2da82a666a321dcc9f5d
2020-05-01 14:20:58 -07:00
..
BUILD
[XLA:Python] Specify a 2MiB stack size for host stream threads.
2020-05-01 14:20:58 -07:00
host_gpu_executor.cc
[XLA:Python] Specify a 2MiB stack size for host stream threads.
2020-05-01 14:20:58 -07:00
host_gpu_executor.h
[XLA:Python] Specify a 2MiB stack size for host stream threads.
2020-05-01 14:20:58 -07:00
host_platform_id.cc
…
host_platform_id.h
…
host_platform.cc
Qualify uses of std::string.
2020-03-25 09:53:25 -07:00
host_platform.h
Qualify uses of std::string.
2020-03-25 09:53:25 -07:00
host_stream_test.cc
…
host_stream.cc
[XLA:Python] Specify a 2MiB stack size for host stream threads.
2020-05-01 14:20:58 -07:00
host_stream.h
[XLA:Python] Specify a 2MiB stack size for host stream threads.
2020-05-01 14:20:58 -07:00
host_timer.cc
…
host_timer.h
…