077b553fda
[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 |
||
|---|---|---|
| .. | ||
| distributed | ||
| BUILD | ||
| cpu_device.cc | ||
| cpu_device.h | ||
| event_pool.cc | ||
| event_pool.h | ||
| gpu_multistream_test.cc | ||
| local_device_state.cc | ||
| local_device_state.h | ||
| nvidia_gpu_device.cc | ||
| nvidia_gpu_device.h | ||
| pjrt_client.cc | ||
| pjrt_client.h | ||
| semaphore_test.cc | ||
| semaphore.cc | ||
| semaphore.h | ||
| tracked_device_buffer_test.cc | ||
| tracked_device_buffer.cc | ||
| tracked_device_buffer.h | ||
| worker_thread.cc | ||
| worker_thread.h | ||