mirror of
https://github.com/intel/compute-runtime.git
synced 2025-12-24 12:23:05 +08:00
30a67618f8
Related-To: NEO-5225 Change-Id: I8cf0aef3ec16314cfb1a787852b6c20ce0f65955 Signed-off-by: Zbigniew Zdanowicz <zbigniew.zdanowicz@intel.com>
45 lines
1.3 KiB
C++
45 lines
1.3 KiB
C++
/*
|
|
* Copyright (C) 2017-2020 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
#include "shared/source/command_stream/device_command_stream.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, DeviceBitfield deviceBitfield);
|
|
virtual ~WddmCommandStreamReceiver();
|
|
|
|
bool flush(BatchBuffer &batchBuffer, ResidencyContainer &allocationsForResidency) override;
|
|
void 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;
|
|
|
|
protected:
|
|
void kmDafLockAllocations(ResidencyContainer &allocationsForResidency);
|
|
|
|
Wddm *wddm;
|
|
COMMAND_BUFFER_HEADER_REC *commandBufferHeader;
|
|
};
|
|
} // namespace NEO
|