Searched refs:debug_allocator_str (Results 1 – 1 of 1) sorted by relevance
46 const char* debug_allocator_str = std::getenv("TF_GPU_ALLOCATOR"); in useCudaMallocAllocator() local47 return debug_allocator_str != nullptr && in useCudaMallocAllocator()48 std::strcmp(debug_allocator_str, "cuda_malloc") == 0; in useCudaMallocAllocator()52 const char* debug_allocator_str = std::getenv("TF_GPU_ALLOCATOR"); in useCudaMemoryGuardAllocator() local53 return debug_allocator_str != nullptr && in useCudaMemoryGuardAllocator()54 std::strcmp(debug_allocator_str, "memory_guard") == 0; in useCudaMemoryGuardAllocator()