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)) {