Searched refs:dev_allocator (Results 1 – 2 of 2) sorted by relevance
733 Allocator* dev_allocator = nullptr; in GetDeviceAndAllocator() local749 dev_allocator = GPUProcessState::singleton()->GetGPUAllocator( in GetDeviceAndAllocator()756 return std::make_pair(cuda_device_id, dev_allocator); in GetDeviceAndAllocator()777 dev_allocator = devices[0]->GetAllocator(alloc_attr); in GetDeviceAndAllocator()778 VLOG(1) << "Using allocator " << dev_allocator->Name() in GetDeviceAndAllocator()784 return std::make_pair(cuda_device_id, dev_allocator); in GetDeviceAndAllocator()
461 auto dev_allocator = in ReportAllocsOnResourceExhausted() local463 AllocStats& dev_allocs_stats = allocs_map[dev_allocator]; in ReportAllocsOnResourceExhausted()