xla icon indicating copy to clipboard operation
xla copied to clipboard

Published 20 hours ago • verified publisher icon openxla

[PJRT:C] Add PJRT_Client_Defragment

Open copybara-service[bot] opened this issue 2 years ago • 0 comments

[PJRT:C] Add PJRT_Client_Defragment

Also adds some helper functions:

  • PJRT_RETURN_STATUS_IF_ERROR (macro)
  • AwaitEvent
  • DefragmentClient
  • GetHostBuffer

[JAX] Modify JaxArrayTest.test_defragment to work on any numbers of devices

copybara-service[bot] avatar Feb 15 '23 01:02 copybara-service[bot]