mirror of
https://github.com/intel/compute-runtime.git
synced 2025-09-15 13:01:45 +08:00
645600d141
We want to return error code to the application instead of aborting when we are not able to make more memory resident. Related-To: NEO-7289 Signed-off-by: Fabian Zwolinski <fabian.zwolinski@intel.com>
48 lines
1.4 KiB
C++
48 lines
1.4 KiB
C++
/*
|
|
* Copyright (C) 2018-2022 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
#include "shared/source/command_stream/device_command_stream.h"
|
|
#include "shared/source/command_stream/submission_status.h"
|
|
|
|
struct COMMAND_BUFFER_HEADER_REC;
|
|
|
|
namespace NEO {
|
|
class GmmPageTableMngr;
|
|
class GraphicsAllocation;
|
|
class WddmMemoryManager;
|
|
class Wddm;
|
|
|
|
template <typename GfxFamily>
|
|
class WddmCommandStreamReceiver : public DeviceCommandStreamReceiver<GfxFamily> {
|
|
typedef DeviceCommandStreamReceiver<GfxFamily> BaseClass;
|
|
|
|
public:
|
|
WddmCommandStreamReceiver(ExecutionEnvironment &executionEnvironment, uint32_t rootDeviceIndex, const DeviceBitfield deviceBitfield);
|
|
~WddmCommandStreamReceiver() override;
|
|
|
|
SubmissionStatus flush(BatchBuffer &batchBuffer, ResidencyContainer &allocationsForResidency) override;
|
|
bool processResidency(const ResidencyContainer &allocationsForResidency, uint32_t handleId) override;
|
|
void processEviction() override;
|
|
bool waitForFlushStamp(FlushStamp &flushStampToWait) override;
|
|
|
|
WddmMemoryManager *getMemoryManager() const;
|
|
Wddm *peekWddm() const {
|
|
return wddm;
|
|
}
|
|
GmmPageTableMngr *createPageTableManager() override;
|
|
|
|
using CommandStreamReceiver::pageTableManager;
|
|
|
|
protected:
|
|
void kmDafLockAllocations(ResidencyContainer &allocationsForResidency);
|
|
|
|
Wddm *wddm;
|
|
COMMAND_BUFFER_HEADER_REC *commandBufferHeader;
|
|
};
|
|
} // namespace NEO
|