Commit f126d1d8 authored by Benjamin Kramer's avatar Benjamin Kramer Committed by TensorFlower Gardener
Browse files

[XLA:GPU] Schedule freeing the tuple address buffer after it's use 

In theory cuMemcpyHtoDAsync can access the memory asynchronously after it went
out of scope. Synchronize releasing the memory with the memcpy completion.

PiperOrigin-RevId: 207686517
parent 7fef41bc
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment