| 32 | namespace tensorflow { |
| 33 | |
| 34 | bool RedzoneCheckDisabled() { |
| 35 | const char* disable_rz_str = std::getenv("TF_DISABLE_RZ_CHECK"); |
| 36 | return disable_rz_str != nullptr && std::strcmp(disable_rz_str, "1") == 0; |
| 37 | } |
| 38 | |
| 39 | se::DeviceMemoryBase WrapRedzoneBestEffort( |
| 40 | se::cuda::RedzoneAllocator* rz_allocator, se::DeviceMemoryBase buffer) { |
no outgoing calls
no test coverage detected