Commit 200f4a20 authored by Justin Lebar's avatar Justin Lebar Committed by TensorFlower Gardener
Browse files

Fix oom_test so that it doesn't try to allocate a giant host buffer when 

run without --config=cuda.  Sadly the best way I could come up with is
pretty hacky.

PiperOrigin-RevId: 195325149
parent 549d63ac
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