xla
Make WaitDeviceOps block until device execution finishes
#4626
Merged

Make WaitDeviceOps block until device execution finishes #4626

JackCaoG merged 9 commits into master from JackCaoG/PJRT_wait_buffer_sync
JackCaoG
JackCaoG JackCaoG requested a review from will-cromar will-cromar 2 years ago
stgpetrovic
stgpetrovic commented on 2023-02-15
will-cromar
will-cromar commented on 2023-02-15
JackCaoG JackCaoG requested a review from stgpetrovic stgpetrovic 2 years ago
JackCaoG JackCaoG requested a review from will-cromar will-cromar 2 years ago
will-cromar
will-cromar approved these changes on 2023-02-16
JackCaoG
JackCaoG
JackCaoG Make WaitDeviceOps block until device execution finishes
dc945429
JackCaoG Add comment
feec68cb
JackCaoG Add test
263df76b
JackCaoG Use shared_mutex instead
37b96a32
JackCaoG Update comment
e1fdd5ab
JackCaoG typo
b2a399b7
JackCaoG Remove assertNotIn in the test since it is too unstbale
73040fac
JackCaoG JackCaoG force pushed from 01e63abe to 73040fac 2 years ago
JackCaoG
JackCaoG handle opbyop
77d39a4b
JackCaoG Update test_operations.py
83e12e52
JackCaoG JackCaoG merged 6d1158cf into master 2 years ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone