[xla:gpu] Create a non-atomically upgradeable reader mutex lock that acquires...
[xla:gpu] Create a non-atomically upgradeable reader mutex lock that acquires and releases the lock via RAII Runtime autotuning needs to be protected by a writer mutex lock. However GpuExecutable runs under a reader mutex lock, so it needs to be upgraded to a writer lock whenever autotuning happens. PiperOrigin-RevId: 510509236
Showing
- tensorflow/compiler/xla/service/gpu/BUILD 21 additions, 0 deletionstensorflow/compiler/xla/service/gpu/BUILD
- tensorflow/compiler/xla/service/gpu/non_atomically_upgradeable_rw_lock.h 95 additions, 0 deletions...iler/xla/service/gpu/non_atomically_upgradeable_rw_lock.h
- tensorflow/compiler/xla/service/gpu/non_atomically_upgradeable_rw_lock_test.cc 45 additions, 0 deletions...la/service/gpu/non_atomically_upgradeable_rw_lock_test.cc
Please register or sign in to comment