Fix a potential memory corruption in PJRT GPU client. Make sure the staging buffer outlives the memcpys to and from the staging buffer.
Fix a potential memory corruption in PJRT GPU client. Make sure the staging buffer outlives the memcpys to and from the staging buffer.
