simple_pool_manager.cc revision 11308
111308Santhony.gutierrez@amd.com/* 211308Santhony.gutierrez@amd.com * Copyright (c) 2015 Advanced Micro Devices, Inc. 311308Santhony.gutierrez@amd.com * All rights reserved. 411308Santhony.gutierrez@amd.com * 511308Santhony.gutierrez@amd.com * For use for simulation and test purposes only 611308Santhony.gutierrez@amd.com * 711308Santhony.gutierrez@amd.com * Redistribution and use in source and binary forms, with or without 811308Santhony.gutierrez@amd.com * modification, are permitted provided that the following conditions are met: 911308Santhony.gutierrez@amd.com * 1011308Santhony.gutierrez@amd.com * 1. Redistributions of source code must retain the above copyright notice, 1111308Santhony.gutierrez@amd.com * this list of conditions and the following disclaimer. 1211308Santhony.gutierrez@amd.com * 1311308Santhony.gutierrez@amd.com * 2. Redistributions in binary form must reproduce the above copyright notice, 1411308Santhony.gutierrez@amd.com * this list of conditions and the following disclaimer in the documentation 1511308Santhony.gutierrez@amd.com * and/or other materials provided with the distribution. 1611308Santhony.gutierrez@amd.com * 1711308Santhony.gutierrez@amd.com * 3. Neither the name of the copyright holder nor the names of its contributors 1811308Santhony.gutierrez@amd.com * may be used to endorse or promote products derived from this software 1911308Santhony.gutierrez@amd.com * without specific prior written permission. 2011308Santhony.gutierrez@amd.com * 2111308Santhony.gutierrez@amd.com * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" 2211308Santhony.gutierrez@amd.com * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 2311308Santhony.gutierrez@amd.com * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 2411308Santhony.gutierrez@amd.com * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE 2511308Santhony.gutierrez@amd.com * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 2611308Santhony.gutierrez@amd.com * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 2711308Santhony.gutierrez@amd.com * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 2811308Santhony.gutierrez@amd.com * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 2911308Santhony.gutierrez@amd.com * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 3011308Santhony.gutierrez@amd.com * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 3111308Santhony.gutierrez@amd.com * POSSIBILITY OF SUCH DAMAGE. 3211308Santhony.gutierrez@amd.com * 3311308Santhony.gutierrez@amd.com * Author: John Kalamatianos 3411308Santhony.gutierrez@amd.com */ 3511308Santhony.gutierrez@amd.com 3611308Santhony.gutierrez@amd.com#include "gpu-compute/simple_pool_manager.hh" 3711308Santhony.gutierrez@amd.com 3811308Santhony.gutierrez@amd.com#include "base/misc.hh" 3911308Santhony.gutierrez@amd.com 4011308Santhony.gutierrez@amd.com// return the min number of elements that the manager can reserve given 4111308Santhony.gutierrez@amd.com// a request for "size" elements 4211308Santhony.gutierrez@amd.comuint32_t 4311308Santhony.gutierrez@amd.comSimplePoolManager::minAllocatedElements(uint32_t size) 4411308Santhony.gutierrez@amd.com{ 4511308Santhony.gutierrez@amd.com fatal_if(size <= 0 || size > poolSize(), "Illegal VGPR region size=%d\n", 4611308Santhony.gutierrez@amd.com size); 4711308Santhony.gutierrez@amd.com 4811308Santhony.gutierrez@amd.com return size % minAllocation() > 0 ? 4911308Santhony.gutierrez@amd.com (minAllocation() - (size % minAllocation())) + size : size; 5011308Santhony.gutierrez@amd.com} 5111308Santhony.gutierrez@amd.com 5211308Santhony.gutierrez@amd.comstd::string 5311308Santhony.gutierrez@amd.comSimplePoolManager::printRegion() 5411308Santhony.gutierrez@amd.com{ 5511308Santhony.gutierrez@amd.com std::string _cout; 5611308Santhony.gutierrez@amd.com if (_reservedGroups == 0) 5711308Santhony.gutierrez@amd.com _cout = "VRF is empty\n"; 5811308Santhony.gutierrez@amd.com else if (_reservedGroups > 0) { 5911308Santhony.gutierrez@amd.com uint32_t reservedEntries = _reservedGroups * _regionSize; 6011308Santhony.gutierrez@amd.com _cout = "VRF reserves " + std::to_string(reservedEntries) + " VGPRs\n"; 6111308Santhony.gutierrez@amd.com } 6211308Santhony.gutierrez@amd.com 6311308Santhony.gutierrez@amd.com return _cout; 6411308Santhony.gutierrez@amd.com} 6511308Santhony.gutierrez@amd.com 6611308Santhony.gutierrez@amd.combool 6711308Santhony.gutierrez@amd.comSimplePoolManager::canAllocate(uint32_t numRegions, uint32_t size) 6811308Santhony.gutierrez@amd.com{ 6911308Santhony.gutierrez@amd.com assert(numRegions * minAllocatedElements(size) <= poolSize()); 7011308Santhony.gutierrez@amd.com 7111308Santhony.gutierrez@amd.com return _reservedGroups == 0; 7211308Santhony.gutierrez@amd.com} 7311308Santhony.gutierrez@amd.com 7411308Santhony.gutierrez@amd.comvoid 7511308Santhony.gutierrez@amd.comSimplePoolManager::freeRegion(uint32_t firstIdx, uint32_t lastIdx) 7611308Santhony.gutierrez@amd.com{ 7711308Santhony.gutierrez@amd.com assert(_reservedGroups > 0); 7811308Santhony.gutierrez@amd.com --_reservedGroups; 7911308Santhony.gutierrez@amd.com 8011308Santhony.gutierrez@amd.com if (!_reservedGroups) 8111308Santhony.gutierrez@amd.com _nxtFreeIdx = 0; 8211308Santhony.gutierrez@amd.com} 8311308Santhony.gutierrez@amd.com 8411308Santhony.gutierrez@amd.comuint32_t 8511308Santhony.gutierrez@amd.comSimplePoolManager::allocateRegion(const uint32_t size, 8611308Santhony.gutierrez@amd.com uint32_t *reservedPoolSize) 8711308Santhony.gutierrez@amd.com{ 8811308Santhony.gutierrez@amd.com uint32_t actualSize = minAllocatedElements(size); 8911308Santhony.gutierrez@amd.com uint32_t startIdx = _nxtFreeIdx; 9011308Santhony.gutierrez@amd.com _nxtFreeIdx += actualSize; 9111308Santhony.gutierrez@amd.com _regionSize = actualSize; 9211308Santhony.gutierrez@amd.com assert(_nxtFreeIdx < poolSize()); 9311308Santhony.gutierrez@amd.com *reservedPoolSize = actualSize; 9411308Santhony.gutierrez@amd.com ++_reservedGroups; 9511308Santhony.gutierrez@amd.com 9611308Santhony.gutierrez@amd.com return startIdx; 9711308Santhony.gutierrez@amd.com} 9811308Santhony.gutierrez@amd.com 9911308Santhony.gutierrez@amd.comuint32_t 10011308Santhony.gutierrez@amd.comSimplePoolManager::regionSize(std::pair<uint32_t, uint32_t> ®ion) 10111308Santhony.gutierrez@amd.com{ 10211308Santhony.gutierrez@amd.com bool wrapAround = (region.first > region.second); 10311308Santhony.gutierrez@amd.com if (!wrapAround) { 10411308Santhony.gutierrez@amd.com return region.second - region.first + 1; 10511308Santhony.gutierrez@amd.com } else { 10611308Santhony.gutierrez@amd.com return region.second + poolSize() - region.first + 1; 10711308Santhony.gutierrez@amd.com } 10811308Santhony.gutierrez@amd.com} 109