47 lines
1.4 KiB
C++
47 lines
1.4 KiB
C++
/*
|
|
* Copyright (C) 2018-2020 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
#include "os_interface/os_context.h"
|
|
#include "os_interface/windows/wddm_residency_controller.h"
|
|
|
|
namespace NEO {
|
|
|
|
class Wddm;
|
|
|
|
struct HardwareQueue {
|
|
D3DKMT_HANDLE handle = 0;
|
|
D3DKMT_HANDLE progressFenceHandle = 0;
|
|
VOID *progressFenceCpuVA = nullptr;
|
|
D3DGPU_VIRTUAL_ADDRESS progressFenceGpuVA = 0;
|
|
};
|
|
|
|
class OsContextWin : public OsContext {
|
|
public:
|
|
OsContextWin() = delete;
|
|
~OsContextWin() override;
|
|
|
|
OsContextWin(Wddm &wddm, uint32_t contextId, DeviceBitfield deviceBitfield,
|
|
aub_stream::EngineType engineType, PreemptionMode preemptionMode, bool lowPriority);
|
|
|
|
D3DKMT_HANDLE getWddmContextHandle() const { return wddmContextHandle; }
|
|
void setWddmContextHandle(D3DKMT_HANDLE wddmContextHandle) { this->wddmContextHandle = wddmContextHandle; }
|
|
HardwareQueue getHwQueue() const { return hardwareQueue; }
|
|
void setHwQueue(HardwareQueue hardwareQueue) { this->hardwareQueue = hardwareQueue; }
|
|
bool isInitialized() const override;
|
|
Wddm *getWddm() const { return &wddm; }
|
|
MOCKABLE_VIRTUAL WddmResidencyController &getResidencyController() { return residencyController; }
|
|
|
|
protected:
|
|
bool initialized = false;
|
|
D3DKMT_HANDLE wddmContextHandle = 0;
|
|
HardwareQueue hardwareQueue;
|
|
Wddm &wddm;
|
|
WddmResidencyController residencyController;
|
|
};
|
|
} // namespace NEO
|