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

[XLA:GPU] Make sure the input for ThenMemcpy stays alive when it executes

asynchronously

By the time memcpy actually executes the source on stack would've gone out of
scope. Also drop an unneeded BlockHostUntilDone from the autotune code.

PiperOrigin-RevId: 211422876
parent c1385b0d
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment