Commit dcccc7cd authored by Peter Hawkins's avatar Peter Hawkins Committed by TensorFlower Gardener
Browse files

[TF:XLA:GPU] Register one XLA_GPU device per device actually present, rather... 

[TF:XLA:GPU] Register one XLA_GPU device per device actually present, rather than unconditionally registering one.

Also set a GpuDeviceContext, since the commented-out code works fine.

PiperOrigin-RevId: 220373651
parent 9abfbf99
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