Commit 4e955be2 authored by Rohan Jain's avatar Rohan Jain Committed by TensorFlower Gardener
Browse files

Fixing a couple of small bugs with the multi device iterator having to deal

with the case when the background thread terminated because the iterator
finished and yet some other requests were coming in.

1. The GetNextFromShard would see an empty buffer and return cancelled instead
of OutOfRange errors
2. On shutdown, we weren't calling all the pending callbacks.

Tested with runs_per_test=5000

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