mirror of
https://github.com/intel/compute-runtime.git
synced 2025-09-15 13:01:45 +08:00
70 lines
3.0 KiB
C++
70 lines
3.0 KiB
C++
/*
|
|
* Copyright (C) 2018-2021 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#include "shared/source/helpers/string.h"
|
|
|
|
#include "opencl/source/gtpin/gtpin_hw_helper.h"
|
|
#include "opencl/source/kernel/kernel.h"
|
|
|
|
#include "hw_cmds.h"
|
|
|
|
namespace NEO {
|
|
|
|
template <typename GfxFamily>
|
|
bool GTPinHwHelperHw<GfxFamily>::addSurfaceState(Kernel *pKernel) {
|
|
using RENDER_SURFACE_STATE = typename GfxFamily::RENDER_SURFACE_STATE;
|
|
using BINDING_TABLE_STATE = typename GfxFamily::BINDING_TABLE_STATE;
|
|
|
|
size_t sshSize = pKernel->getSurfaceStateHeapSize();
|
|
if ((sshSize == 0) || pKernel->isParentKernel) {
|
|
// Kernels which do not use SSH or use Execution Model are not supported (yet)
|
|
return false;
|
|
}
|
|
size_t ssSize = sizeof(RENDER_SURFACE_STATE);
|
|
size_t btsSize = sizeof(BINDING_TABLE_STATE);
|
|
size_t sizeToEnlarge = ssSize + btsSize;
|
|
size_t currBTOffset = pKernel->getBindingTableOffset();
|
|
size_t currSurfaceStateSize = currBTOffset;
|
|
char *pSsh = static_cast<char *>(pKernel->getSurfaceStateHeap());
|
|
char *pNewSsh = new char[sshSize + sizeToEnlarge];
|
|
memcpy_s(pNewSsh, sshSize + sizeToEnlarge, pSsh, currSurfaceStateSize);
|
|
RENDER_SURFACE_STATE *pSS = reinterpret_cast<RENDER_SURFACE_STATE *>(pNewSsh + currSurfaceStateSize);
|
|
*pSS = GfxFamily::cmdInitRenderSurfaceState;
|
|
size_t newSurfaceStateSize = currSurfaceStateSize + ssSize;
|
|
size_t currBTCount = pKernel->getNumberOfBindingTableStates();
|
|
memcpy_s(pNewSsh + newSurfaceStateSize, sshSize + sizeToEnlarge - newSurfaceStateSize, pSsh + currBTOffset, currBTCount * btsSize);
|
|
BINDING_TABLE_STATE *pNewBTS = reinterpret_cast<BINDING_TABLE_STATE *>(pNewSsh + newSurfaceStateSize + currBTCount * btsSize);
|
|
*pNewBTS = GfxFamily::cmdInitBindingTableState;
|
|
pNewBTS->setSurfaceStatePointer((uint64_t)currBTOffset);
|
|
pKernel->resizeSurfaceStateHeap(pNewSsh, sshSize + sizeToEnlarge, currBTCount + 1, newSurfaceStateSize);
|
|
return true;
|
|
}
|
|
|
|
template <typename GfxFamily>
|
|
void *GTPinHwHelperHw<GfxFamily>::getSurfaceState(Kernel *pKernel, size_t bti) {
|
|
using BINDING_TABLE_STATE = typename GfxFamily::BINDING_TABLE_STATE;
|
|
|
|
if ((nullptr == pKernel->getSurfaceStateHeap()) || (bti >= pKernel->getNumberOfBindingTableStates())) {
|
|
return nullptr;
|
|
}
|
|
auto *pBts = reinterpret_cast<BINDING_TABLE_STATE *>(ptrOffset(pKernel->getSurfaceStateHeap(), (pKernel->getBindingTableOffset() + bti * sizeof(BINDING_TABLE_STATE))));
|
|
auto pSurfaceState = ptrOffset(pKernel->getSurfaceStateHeap(), pBts->getSurfaceStatePointer());
|
|
return pSurfaceState;
|
|
}
|
|
|
|
template <typename GfxFamily>
|
|
bool GTPinHwHelperHw<GfxFamily>::canUseSharedAllocation(const HardwareInfo &hwInfo) const {
|
|
bool canUseSharedAllocation = false;
|
|
if (DebugManager.flags.GTPinAllocateBufferInSharedMemory.get() != -1) {
|
|
canUseSharedAllocation = !!DebugManager.flags.GTPinAllocateBufferInSharedMemory.get();
|
|
}
|
|
canUseSharedAllocation &= hwInfo.capabilityTable.ftrSvm;
|
|
return canUseSharedAllocation;
|
|
}
|
|
|
|
} // namespace NEO
|