Searched refs:debug_allocator_str (Results 1 – 1 of 1) sorted by relevance
47 const char* debug_allocator_str = std::getenv("TF_GPU_ALLOCATOR"); in useCudaMallocAllocator() local48 return debug_allocator_str != nullptr && in useCudaMallocAllocator()49 std::strcmp(debug_allocator_str, "cuda_malloc") == 0; in useCudaMallocAllocator()53 const char* debug_allocator_str = std::getenv("TF_GPU_ALLOCATOR"); in useCudaMemoryGuardAllocator() local54 return debug_allocator_str != nullptr && in useCudaMemoryGuardAllocator()55 std::strcmp(debug_allocator_str, "memory_guard") == 0; in useCudaMemoryGuardAllocator()