Searched refs:dev_allocator (Results 1 – 2 of 2) sorted by relevance
645 Allocator* dev_allocator = nullptr; in GetDeviceAndAllocator() local658 dev_allocator = GPUProcessState::singleton()->GetGPUAllocator( in GetDeviceAndAllocator()661 return std::make_pair(cuda_device_id, dev_allocator); in GetDeviceAndAllocator()682 dev_allocator = devices[0]->GetAllocator(alloc_attr); in GetDeviceAndAllocator()683 VLOG(1) << "Using allocator " << dev_allocator->Name() in GetDeviceAndAllocator()689 return std::make_pair(cuda_device_id, dev_allocator); in GetDeviceAndAllocator()
459 auto dev_allocator = in ReportAllocsOnResourceExhausted() local461 AllocStats& dev_allocs_stats = allocs_map[dev_allocator]; in ReportAllocsOnResourceExhausted()