Searched refs:dev_allocator (Results 1 – 2 of 2) sorted by relevance
634 Allocator* dev_allocator = nullptr; in GetDeviceAndAllocator() local647 dev_allocator = GPUProcessState::singleton()->GetGPUAllocator( in GetDeviceAndAllocator()650 return std::make_pair(cuda_device_id, dev_allocator); in GetDeviceAndAllocator()671 dev_allocator = devices[0]->GetAllocator(alloc_attr); in GetDeviceAndAllocator()672 VLOG(1) << "Using allocator " << dev_allocator->Name() in GetDeviceAndAllocator()678 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()