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

In BaseGPUDeviceFactory::CreateDevices() move the check for no GPU devices a little earlier. 

This saves ~83MiB of GPU DRAM for each (unused) GPU device, and some CPU heap memory allocated
by libcuda.

PiperOrigin-RevId: 210803477
parent 62bcfd96
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