Lines Matching defs:MemoryPool
262 AMDGPUMemoryPoolTy(hsa_amd_memory_pool_t MemoryPool)
263 : MemoryPool(MemoryPool), GlobalFlags(0) {}
277 hsa_amd_memory_pool_t get() const { return MemoryPool; }
306 hsa_amd_memory_pool_allocate(MemoryPool, Size, 0, PtrStorage);
320 Agent, MemoryPool, HSA_AMD_AGENT_MEMORY_POOL_INFO_ACCESS, &Access))
352 Status = hsa_amd_memory_pool_get_info(MemoryPool, Kind, &Value);
358 return hsa_amd_memory_pool_get_info(MemoryPool, Kind, &Value);
367 hsa_amd_agent_memory_pool_get_info(Agent, MemoryPool, Kind, &Value);
374 hsa_amd_memory_pool_t MemoryPool;
390 : Plugin(Plugin), MemoryPool(nullptr), MemoryManager(nullptr) {}
393 Error init(AMDGPUMemoryPoolTy &MemoryPool) {
396 this->MemoryPool = &MemoryPool;
441 if (auto Err = MemoryPool->deallocate(TgtPtr)) {
452 AMDGPUMemoryPoolTy *MemoryPool;
1742 for (AMDGPUMemoryPoolTy *MemoryPool : AllMemoryPools) {
1744 Error Err = MemoryPool->init();
1748 if (!MemoryPool->isGlobal())
1752 if (MemoryPool->isFineGrained()) {
1753 FineGrainedMemoryPools.push_back(MemoryPool);
1754 if (MemoryPool->supportsKernelArgs())
1755 ArgsMemoryPools.push_back(MemoryPool);
1756 } else if (MemoryPool->isCoarseGrained()) {
1757 CoarseGrainedMemoryPools.push_back(MemoryPool);
1842 AMDGPUMemoryPoolTy *MemoryPool =
1844 AllMemoryPools.push_back(MemoryPool);
2207 AMDGPUMemoryPoolTy *MemoryPool = nullptr;
2212 MemoryPool = CoarseGrainedMemoryPools[0];
2215 MemoryPool = &HostDevice.getFineGrainedMemoryPool();
2218 MemoryPool = &HostDevice.getFineGrainedMemoryPool();
2222 if (!MemoryPool) {
2227 if (Error Err = MemoryPool->deallocate(TgtPtr)) {
2807 AMDGPUMemoryPoolTy *MemoryPool =
2809 new (MemoryPool) AMDGPUMemoryPoolTy(HSAMemoryPool);
2810 AllMemoryPools.push_back(MemoryPool);
3485 if (auto Err = MemoryPool->allocate(Size, &Ptr)) {
3495 [&](hsa_agent_t Agent) { return MemoryPool->canAccess(Agent); });
3498 if (auto Err = MemoryPool->enableAccess(Ptr, Size, Agents)) {
3510 AMDGPUMemoryPoolTy *MemoryPool = nullptr;
3515 MemoryPool = CoarseGrainedMemoryPools[0];
3518 MemoryPool = &HostDevice.getFineGrainedMemoryPool();
3521 MemoryPool = &HostDevice.getFineGrainedMemoryPool();
3525 if (!MemoryPool) {
3532 if (Error Err = MemoryPool->allocate(Size, &Alloc)) {
3544 return MemoryPool->canAccess(Agent);
3548 if (auto Err = MemoryPool->enableAccess(Alloc, Size, Agents)) {