Commit a2d2ac24 authored by Sanjoy Das's avatar Sanjoy Das Committed by TensorFlower Gardener
Browse files

Fix a race condition in XlaLocalLaunchBase. 

XlaLocalLaunchBase was modifying platform_id_ without a lock which is racy
because the same OpKernel can be execute concurrently.  Fix this by inferring
platform_id_ in the kernel constructor.

While at it, make use_multiple_streams_ and xla_device_metadata_ member
variables also.

PiperOrigin-RevId: 210751494
parent 3df3d078
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