Commit 0898ee30 authored by Benoit Steiner's avatar Benoit Steiner Committed by Gunhan Gulsoy
Browse files

Use optimized ops to handle GPU memory swapping: this avoids the need for 2 

pairs of extra _send/_recv nodes which speeds things up a bit. This also
ensures that performance doesn't depend on the recv scheduling built in TF,
which isn't always optimal.

PiperOrigin-RevId: 187057831
parent 5caeb37e
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