experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
rei/fork-r2.3
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 [StreamExecutor] Rename ::perftools::gputools -> ::stream_executor, part 1. 2018-04-17 14:28:51 -07:00
host_platform_id.h [StreamExecutor] Rename ::perftools::gputools -> ::stream_executor, part 1. 2018-04-17 14:28:51 -07:00
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 [SE] Fix FIFO ordering of HostStream. 2019-05-06 18:11:03 -07:00
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 [StreamExecutor] Rename ::perftools::gputools -> ::stream_executor, part 1. 2018-04-17 14:28:51 -07:00
host_timer.h [StreamExecutor] Rename ::perftools::gputools -> ::stream_executor, part 1. 2018-04-17 14:28:51 -07:00