Commit cf1e86da authored by Adrian Kuegel's avatar Adrian Kuegel Committed by TensorFlower Gardener
Browse files

Don't try to call cuMemAlloc with size 0.

This will result in a failure, so we can just immediately return nullptr.
The caller code currently already needs to accept nullptr as a valid result if
size == 0, and only treat it as an error if size > 0.
PiperOrigin-RevId: 234467182
parent bd46f91a
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment