xla
Re-implement `ComputationClient::WaitDeviceOps`
#5811
Merged

Commits
  • Replace device_locks_
    will-cromar committed 2 years ago
  • block new ops while waiting
    will-cromar committed 2 years ago
  • get lock before notifying cv
    will-cromar committed 2 years ago
  • Hold onto device locks until wait is complete
    will-cromar committed 2 years ago
  • formatting
    will-cromar committed 2 years ago
  • shared_ptr -> unique_ptr
    will-cromar committed 2 years ago
  • remove lock_device
    will-cromar committed 2 years ago
  • construct counter inplace
    will-cromar committed 2 years ago
  • improve logging
    will-cromar committed 2 years ago
  • make operation_tracker private
    will-cromar committed 2 years ago
  • formatting
    will-cromar committed 2 years ago
  • Comments
    will-cromar committed 2 years ago
  • absl span
    will-cromar committed 2 years ago
  • address review comments
    will-cromar committed 2 years ago
  • remove comment
    will-cromar committed 2 years ago
  • actually make a vector for `returned_futures`
    will-cromar committed 2 years ago
  • Revert "address review comments"
    will-cromar committed 2 years ago
  • naming
    will-cromar committed 2 years ago
  • correct comment
    will-cromar committed 2 years ago
Loading