mirror of
https://github.com/intel/compute-runtime.git
synced 2025-12-25 13:33:02 +08:00
49d3c39fae
Related-To: NEO-4338 Change-Id: Ic858a9324e5f892532d39c98a4029df9d2a64e46 Signed-off-by: Zbigniew Zdanowicz <zbigniew.zdanowicz@intel.com>
112 lines
3.7 KiB
C++
112 lines
3.7 KiB
C++
/*
|
|
* Copyright (C) 2020 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
#include "shared/source/direct_submission/direct_submission_hw.h"
|
|
#include "shared/source/memory_manager/graphics_allocation.h"
|
|
|
|
namespace NEO {
|
|
|
|
template <typename GfxFamily, typename Dispatcher>
|
|
struct MockDirectSubmissionHw : public DirectSubmissionHw<GfxFamily, Dispatcher> {
|
|
using BaseClass = DirectSubmissionHw<GfxFamily, Dispatcher>;
|
|
using BaseClass::allocateResources;
|
|
using BaseClass::completionRingBuffers;
|
|
using BaseClass::cpuCachelineFlush;
|
|
using BaseClass::currentQueueWorkCount;
|
|
using BaseClass::currentRingBuffer;
|
|
using BaseClass::deallocateResources;
|
|
using BaseClass::device;
|
|
using BaseClass::DirectSubmissionHw;
|
|
using BaseClass::disableCpuCacheFlush;
|
|
using BaseClass::dispatchEndingSection;
|
|
using BaseClass::dispatchFlushSection;
|
|
using BaseClass::dispatchSemaphoreSection;
|
|
using BaseClass::dispatchStartSection;
|
|
using BaseClass::dispatchSwitchRingBufferSection;
|
|
using BaseClass::dispatchTagUpdateSection;
|
|
using BaseClass::dispatchWorkloadSection;
|
|
using BaseClass::getCommandBufferPositionGpuAddress;
|
|
using BaseClass::getSizeDispatch;
|
|
using BaseClass::getSizeEnd;
|
|
using BaseClass::getSizeEndingSection;
|
|
using BaseClass::getSizeFlushSection;
|
|
using BaseClass::getSizeSemaphoreSection;
|
|
using BaseClass::getSizeStartSection;
|
|
using BaseClass::getSizeStoraDataSection;
|
|
using BaseClass::getSizeSwitchRingBufferSection;
|
|
using BaseClass::getSizeTagUpdateSection;
|
|
using BaseClass::hwInfo;
|
|
using BaseClass::osContext;
|
|
using BaseClass::ringBuffer;
|
|
using BaseClass::ringBuffer2;
|
|
using BaseClass::ringCommandStream;
|
|
using BaseClass::ringStart;
|
|
using BaseClass::semaphoreData;
|
|
using BaseClass::semaphoreGpuVa;
|
|
using BaseClass::semaphorePtr;
|
|
using BaseClass::semaphores;
|
|
using BaseClass::setReturnAddress;
|
|
using BaseClass::stopRingBuffer;
|
|
using BaseClass::switchRingBuffersAllocations;
|
|
using typename BaseClass::RingBufferUse;
|
|
|
|
~MockDirectSubmissionHw() override {
|
|
if (ringStart) {
|
|
stopRingBuffer();
|
|
}
|
|
deallocateResources();
|
|
}
|
|
|
|
bool allocateOsResources(DirectSubmissionAllocations &allocations) override {
|
|
return allocateOsResourcesReturn;
|
|
}
|
|
|
|
bool submit(uint64_t gpuAddress, size_t size) override {
|
|
submitGpuAddress = gpuAddress;
|
|
submitSize = size;
|
|
submitCount++;
|
|
return submitReturn;
|
|
}
|
|
|
|
bool handleResidency() override {
|
|
handleResidencyCount++;
|
|
return handleResidencyReturn;
|
|
}
|
|
|
|
uint64_t switchRingBuffers() override {
|
|
GraphicsAllocation *nextRingBuffer = switchRingBuffersAllocations();
|
|
uint64_t currentBufferGpuVa = getCommandBufferPositionGpuAddress(ringCommandStream.getSpace(0));
|
|
|
|
ringCommandStream.replaceBuffer(nextRingBuffer->getUnderlyingBuffer(), ringCommandStream.getMaxAvailableSpace());
|
|
ringCommandStream.replaceGraphicsAllocation(nextRingBuffer);
|
|
|
|
return currentBufferGpuVa;
|
|
}
|
|
|
|
uint64_t updateTagValue() override {
|
|
return updateTagValueReturn;
|
|
}
|
|
|
|
void getTagAddressValue(TagData &tagData) override {
|
|
tagData.tagAddress = tagAddressSetValue;
|
|
tagData.tagValue = tagValueSetValue;
|
|
}
|
|
|
|
bool allocateOsResourcesReturn = true;
|
|
bool submitReturn = true;
|
|
bool handleResidencyReturn = true;
|
|
uint32_t submitCount = 0u;
|
|
uint32_t handleResidencyCount = 0u;
|
|
size_t submitSize = 0u;
|
|
uint64_t updateTagValueReturn = 1ull;
|
|
uint64_t tagAddressSetValue = MemoryConstants::pageSize;
|
|
uint64_t tagValueSetValue = 1ull;
|
|
uint64_t submitGpuAddress = 0ull;
|
|
};
|
|
} // namespace NEO
|