48 lines
1.4 KiB
C++
48 lines
1.4 KiB
C++
/*
|
|
* Copyright (C) 2017-2019 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
#include "runtime/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);
|
|
virtual ~WddmCommandStreamReceiver();
|
|
|
|
FlushStamp flush(BatchBuffer &batchBuffer, ResidencyContainer &allocationsForResidency) override;
|
|
void makeResident(GraphicsAllocation &gfxAllocation) override;
|
|
void processResidency(const ResidencyContainer &allocationsForResidency) override;
|
|
void processEviction() override;
|
|
bool waitForFlushStamp(FlushStamp &flushStampToWait) override;
|
|
|
|
WddmMemoryManager *getMemoryManager() const;
|
|
Wddm *peekWddm() const {
|
|
return wddm;
|
|
}
|
|
GmmPageTableMngr *createPageTableManager() override;
|
|
|
|
protected:
|
|
void initPageTableManagerRegisters(LinearStream &csr) override;
|
|
void kmDafLockAllocations(ResidencyContainer &allocationsForResidency);
|
|
|
|
Wddm *wddm;
|
|
COMMAND_BUFFER_HEADER_REC *commandBufferHeader;
|
|
bool pageTableManagerInitialized = false;
|
|
};
|
|
} // namespace NEO
|