refactor: correct expectations in level zero tests if heapless enabled 3/n

Related-To: NEO-10641
Signed-off-by: Kamil Kopryk <kamil.kopryk@intel.com>
This commit is contained in:
Kamil Kopryk
2024-09-03 09:38:57 +00:00
committed by Compute-Runtime-Automation
parent 7adfa344c0
commit 6d7e2760dc
8 changed files with 171 additions and 73 deletions

View File

@@ -254,6 +254,11 @@ class UltCommandStreamReceiver : public CommandStreamReceiverHw<GfxFamily>, publ
return BaseClass::initializeDeviceWithFirstSubmission(device);
}
void programHeaplessStateProlog(Device &device, LinearStream &commandStream) override {
this->commandStreamHeaplessStateInit = &commandStream;
return BaseClass::programHeaplessStateProlog(device, commandStream);
}
bool writeMemory(GraphicsAllocation &gfxAllocation, bool isChunkCopy, uint64_t gpuVaChunkOffset, size_t chunkSize) override {
writeMemoryParams.totalCallCount++;
if (isChunkCopy) {
@@ -541,6 +546,8 @@ class UltCommandStreamReceiver : public CommandStreamReceiverHw<GfxFamily>, publ
TaskCountType flushBcsTaskReturnValue{};
LinearStream *lastFlushedCommandStream = nullptr;
LinearStream *commandStreamHeaplessStateInit = nullptr;
const IndirectHeap *recordedSsh = nullptr;
std::mutex mutex;