Commit e60c87c9 authored by Guangda Lai's avatar Guangda Lai Committed by TensorFlower Gardener
Browse files

Fix GPUDebugAllocator and GPUNanResetAllocator: 

1. Eliminate the checks when the pointer is nullptr.
2. Fix nan array size. Some application can produce a zero size which would
   cause out-of-range access error.

PiperOrigin-RevId: 192186224
parent 0bf3476f
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