MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / RedzoneCheckDisabled

Function RedzoneCheckDisabled

tensorflow/core/kernels/gpu_utils.cc:34–37  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

32namespace tensorflow {
33
34bool 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
39se::DeviceMemoryBase WrapRedzoneBestEffort(
40 se::cuda::RedzoneAllocator* rz_allocator, se::DeviceMemoryBase buffer) {

Callers 9

operator()Method · 0.85
ComputeMethod · 0.85
WrapRedzoneBestEffortFunction · 0.85
CheckRedzonesFunction · 0.85
operator()Method · 0.85
launchMethod · 0.85
operator()Method · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected