Commit fccaac3d authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

Force GPU device objects that refer to the same physical card using the 

same stream id to use the same cuda stream objects. This avoids confusing
the per-device memory allocator in ways that cause memory corruption.

Fixes https://github.com/tensorflow/serving/issues/335.

PiperOrigin-RevId: 157258318
parent 2ff1d7bf
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