[Crash fix] Fix cudaMallocAsync crashes.#49173

Merged

copybara-service[bot] merged 10 commits intotensorflow:masterfrom

nouiz:upstream-cuda_malloc_async_fixes

Jun 17, 2021

Commits on Jun 8, 2021

[Crash fix] Other part of TF need the stats for GPU allocator.
nouiz
committed
[Crash fix] Handle correctly the passed stream.
nouiz
committed
Add a cudaMallocAsync test
nouiz
committed
Use a simpler static_cast
nouiz
committed
Make the new options experimental and add it to the golden list of public API.
nouiz
committed
Use the new option
nouiz
committed
Test that the new allocator is instantiated, not that there is no crash.
nouiz
committed
Variable/function renaming. Also make a counter atomic to be more future proof. The current initialization code in TF instantiate then serially.
nouiz
committed
Fix a segfaul introduced in 76e61092a7aa8958ea190453d1c51a119de3e189. sub_allocator is owned by gpu_bfc_allocator. deleting it early doesn't any real benefit. So lets the normal clean up do its work.
nouiz
committed
Replace a CHECK to EXPECT_NE.
nouiz
committed