xla
6e661304 - Use upstream XLA concurrency utilities (#5799)

Commit

1 year ago

Use upstream XLA concurrency utilities (#5799) * Use TSL threadpool * remove multiwait * fix test build * Move threadpool namespace * formatting * fix test build * Use BlockingCounter

References

#5799 - Use upstream XLA concurrency utilities

Author

will-cromar

Parents

a3bbbb1e

Files17

test/cpp
- BUILD
- test_replication.cpp
torch_xla/csrc
- BUILD
- init_python_bindings.cpp
- runtime
  - BUILD
  - multi_wait.cc
  - multi_wait.h
  - pjrt_computation_client.cc
  - thread_pool.cc
  - thread_pool.h
- tensor.cpp
- tensor_util.cpp
- thread_pool.cc
- thread_pool.h
- xla_graph_executor.cpp
- xla_graph_executor.h
- xla_sharding_util.cpp

xla 6e661304 - Use upstream XLA concurrency utilities (#5799)

xla
6e661304 - Use upstream XLA concurrency utilities (#5799)