mirror of
https://github.com/intel/compute-runtime.git
synced 2025-09-15 13:01:45 +08:00
Move wddm_fixture.h and releted files from opencl to shared
Related-To: NEO-5894 Signed-off-by: Piotr Obst <piotr.obst@intel.com>
This commit is contained in:
Piotr Obst
committed by
Compute-Runtime-Automation
Compute-Runtime-Automation
parent
0f072a93e0
commit
acf3e58a64
188
shared/test/common/mocks/mock_wddm.h
Normal file
188
shared/test/common/mocks/mock_wddm.h
Normal file
@ -0,0 +1,188 @@
|
||||
/*
|
||||
* Copyright (C) 2018-2021 Intel Corporation
|
||||
*
|
||||
* SPDX-License-Identifier: MIT
|
||||
*
|
||||
*/
|
||||
|
||||
#pragma once
|
||||
|
||||
#include "shared/source/helpers/constants.h"
|
||||
#include "shared/source/helpers/hw_info.h"
|
||||
#include "shared/source/memory_manager/host_ptr_defines.h"
|
||||
#include "shared/source/os_interface/windows/wddm/wddm.h"
|
||||
#include "shared/source/os_interface/windows/wddm_residency_allocations_container.h"
|
||||
#include "shared/source/os_interface/windows/windows_defs.h"
|
||||
#include "shared/test/common/mocks/wddm_mock_helpers.h"
|
||||
|
||||
#include "gmock/gmock.h"
|
||||
|
||||
#include <set>
|
||||
#include <vector>
|
||||
|
||||
namespace NEO {
|
||||
class GraphicsAllocation;
|
||||
|
||||
constexpr auto virtualAllocAddress = is64bit ? 0x7FFFF0000000 : 0xFF000000;
|
||||
|
||||
class WddmMock : public Wddm {
|
||||
public:
|
||||
using Wddm::adapterBDF;
|
||||
using Wddm::createPagingFenceLogger;
|
||||
using Wddm::currentPagingFenceValue;
|
||||
using Wddm::dedicatedVideoMemory;
|
||||
using Wddm::device;
|
||||
using Wddm::deviceRegistryPath;
|
||||
using Wddm::enablePreemptionRegValue;
|
||||
using Wddm::featureTable;
|
||||
using Wddm::getSystemInfo;
|
||||
using Wddm::gmmMemory;
|
||||
using Wddm::hwDeviceId;
|
||||
using Wddm::mapGpuVirtualAddress;
|
||||
using Wddm::minAddress;
|
||||
using Wddm::pagingFenceAddress;
|
||||
using Wddm::pagingQueue;
|
||||
using Wddm::residencyLogger;
|
||||
using Wddm::rootDeviceEnvironment;
|
||||
using Wddm::temporaryResources;
|
||||
using Wddm::timestampFrequency;
|
||||
using Wddm::wddmInterface;
|
||||
|
||||
WddmMock(std::unique_ptr<HwDeviceIdWddm> hwDeviceId, RootDeviceEnvironment &rootDeviceEnvironment) : Wddm(std::move(hwDeviceId), rootDeviceEnvironment) {}
|
||||
WddmMock(RootDeviceEnvironment &rootDeviceEnvironment);
|
||||
~WddmMock();
|
||||
|
||||
bool makeResident(const D3DKMT_HANDLE *handles, uint32_t count, bool cantTrimFurther, uint64_t *numberOfBytesToTrim, size_t totalSize) override;
|
||||
bool evict(const D3DKMT_HANDLE *handles, uint32_t num, uint64_t &sizeToTrim) override;
|
||||
bool mapGpuVirtualAddress(Gmm *gmm, D3DKMT_HANDLE handle, D3DGPU_VIRTUAL_ADDRESS minimumAddress, D3DGPU_VIRTUAL_ADDRESS maximumAddress, D3DGPU_VIRTUAL_ADDRESS preferredAddress, D3DGPU_VIRTUAL_ADDRESS &gpuPtr) override;
|
||||
bool mapGpuVirtualAddress(WddmAllocation *allocation);
|
||||
bool freeGpuVirtualAddress(D3DGPU_VIRTUAL_ADDRESS &gpuPtr, uint64_t size) override;
|
||||
NTSTATUS createAllocation(const void *alignedCpuPtr, const Gmm *gmm, D3DKMT_HANDLE &outHandle, D3DKMT_HANDLE &outResource, D3DKMT_HANDLE *outSharedHandle) override;
|
||||
bool createAllocation64k(const Gmm *gmm, D3DKMT_HANDLE &outHandle) override;
|
||||
bool destroyAllocations(const D3DKMT_HANDLE *handles, uint32_t allocationCount, D3DKMT_HANDLE resourceHandle) override;
|
||||
|
||||
NTSTATUS createAllocation(WddmAllocation *wddmAllocation);
|
||||
bool createAllocation64k(WddmAllocation *wddmAllocation);
|
||||
bool destroyAllocation(WddmAllocation *alloc, OsContextWin *osContext);
|
||||
bool openSharedHandle(D3DKMT_HANDLE handle, WddmAllocation *alloc) override;
|
||||
bool createContext(OsContextWin &osContext) override;
|
||||
void applyAdditionalContextFlags(CREATECONTEXT_PVTDATA &privateData, OsContextWin &osContext, const HardwareInfo &hwInfo) override;
|
||||
bool destroyContext(D3DKMT_HANDLE context) override;
|
||||
bool queryAdapterInfo() override;
|
||||
bool submit(uint64_t commandBuffer, size_t size, void *commandHeader, WddmSubmitArguments &submitArguments) override;
|
||||
bool waitOnGPU(D3DKMT_HANDLE context) override;
|
||||
void *lockResource(const D3DKMT_HANDLE &handle, bool applyMakeResidentPriorToLock, size_t size) override;
|
||||
void unlockResource(const D3DKMT_HANDLE &handle) override;
|
||||
void kmDafLock(D3DKMT_HANDLE handle) override;
|
||||
bool isKmDafEnabled() const override;
|
||||
void setKmDafEnabled(bool state);
|
||||
void setHwContextId(unsigned long hwContextId);
|
||||
void setHeap32(uint64_t base, uint64_t size);
|
||||
GMM_GFX_PARTITIONING *getGfxPartitionPtr();
|
||||
bool waitFromCpu(uint64_t lastFenceValue, const MonitoredFence &monitoredFence) override;
|
||||
void *virtualAlloc(void *inPtr, size_t size, unsigned long flags, unsigned long type) override;
|
||||
int virtualFree(void *ptr, size_t size, unsigned long flags) override;
|
||||
void releaseReservedAddress(void *reservedAddress) override;
|
||||
VOID *registerTrimCallback(PFND3DKMT_TRIMNOTIFICATIONCALLBACK callback, WddmResidencyController &residencyController) override;
|
||||
D3DGPU_VIRTUAL_ADDRESS reserveGpuVirtualAddress(D3DGPU_VIRTUAL_ADDRESS minimumAddress, D3DGPU_VIRTUAL_ADDRESS maximumAddress, D3DGPU_SIZE_T size) override;
|
||||
bool reserveValidAddressRange(size_t size, void *&reservedMem);
|
||||
PLATFORM *getGfxPlatform() { return gfxPlatform.get(); }
|
||||
uint64_t *getPagingFenceAddress() override;
|
||||
void waitOnPagingFenceFromCpu() override;
|
||||
void createPagingFenceLogger() override;
|
||||
bool verifyAdapterLuid(LUID adapterLuid) const override {
|
||||
if (callBaseVerifyAdapterLuid) {
|
||||
return Wddm::verifyAdapterLuid(adapterLuid);
|
||||
}
|
||||
return verifyAdapterLuidReturnValue;
|
||||
}
|
||||
bool setAllocationPriority(const D3DKMT_HANDLE *handles, uint32_t allocationCount, uint32_t priority) override;
|
||||
|
||||
bool configureDeviceAddressSpace() {
|
||||
configureDeviceAddressSpaceResult.called++;
|
||||
//create context cant be called before configureDeviceAddressSpace
|
||||
if (createContextResult.called > 0) {
|
||||
return configureDeviceAddressSpaceResult.success = false;
|
||||
} else {
|
||||
return configureDeviceAddressSpaceResult.success = Wddm::configureDeviceAddressSpace();
|
||||
}
|
||||
}
|
||||
|
||||
uint32_t counterVerifyNTHandle = 0;
|
||||
uint32_t counterVerifySharedHandle = 0;
|
||||
bool verifyNTHandle(HANDLE handle) override {
|
||||
++counterVerifyNTHandle;
|
||||
return Wddm::verifyNTHandle(handle);
|
||||
}
|
||||
bool verifySharedHandle(D3DKMT_HANDLE osHandle) override {
|
||||
++counterVerifySharedHandle;
|
||||
return Wddm::verifySharedHandle(osHandle);
|
||||
}
|
||||
|
||||
bool isShutdownInProgress() override {
|
||||
return shutdownStatus;
|
||||
};
|
||||
|
||||
void resetGdi(Gdi *gdi);
|
||||
|
||||
WddmMockHelpers::MakeResidentCall makeResidentResult;
|
||||
WddmMockHelpers::CallResult makeNonResidentResult;
|
||||
WddmMockHelpers::CallResult mapGpuVirtualAddressResult;
|
||||
WddmMockHelpers::FreeGpuVirtualAddressCall freeGpuVirtualAddressResult;
|
||||
WddmMockHelpers::CallResult createAllocationResult;
|
||||
WddmMockHelpers::CallResult destroyAllocationResult;
|
||||
WddmMockHelpers::CallResult destroyContextResult;
|
||||
WddmMockHelpers::CallResult queryAdapterInfoResult;
|
||||
WddmMockHelpers::SubmitResult submitResult;
|
||||
WddmMockHelpers::CallResult waitOnGPUResult;
|
||||
WddmMockHelpers::CallResult configureDeviceAddressSpaceResult;
|
||||
WddmMockHelpers::CallResult createContextResult;
|
||||
WddmMockHelpers::CallResult applyAdditionalContextFlagsResult;
|
||||
WddmMockHelpers::CallResult lockResult;
|
||||
WddmMockHelpers::CallResult unlockResult;
|
||||
WddmMockHelpers::KmDafLockCall kmDafLockResult;
|
||||
WddmMockHelpers::WaitFromCpuResult waitFromCpuResult;
|
||||
WddmMockHelpers::CallResult releaseReservedAddressResult;
|
||||
WddmMockHelpers::CallResult reserveValidAddressRangeResult;
|
||||
WddmMockHelpers::CallResult registerTrimCallbackResult;
|
||||
WddmMockHelpers::CallResult getPagingFenceAddressResult;
|
||||
WddmMockHelpers::CallResult reserveGpuVirtualAddressResult;
|
||||
WddmMockHelpers::CallResult waitOnPagingFenceFromCpuResult;
|
||||
WddmMockHelpers::CallResult setAllocationPriorityResult;
|
||||
|
||||
NTSTATUS createAllocationStatus = STATUS_SUCCESS;
|
||||
bool verifyAdapterLuidReturnValue = true;
|
||||
bool callBaseVerifyAdapterLuid = false;
|
||||
bool mapGpuVaStatus = true;
|
||||
bool callBaseDestroyAllocations = true;
|
||||
bool failOpenSharedHandle = false;
|
||||
bool callBaseMapGpuVa = true;
|
||||
std::set<void *> reservedAddresses;
|
||||
uintptr_t virtualAllocAddress = NEO::windowsMinAddress;
|
||||
bool kmDafEnabled = false;
|
||||
uint64_t mockPagingFence = 0u;
|
||||
bool makeResidentStatus = true;
|
||||
bool callBaseMakeResident = true;
|
||||
bool callBaseCreatePagingLogger = true;
|
||||
bool shutdownStatus = false;
|
||||
bool callBaseSetAllocationPriority = true;
|
||||
};
|
||||
|
||||
struct GmockWddm : WddmMock {
|
||||
GmockWddm(RootDeviceEnvironment &rootDeviceEnvironment);
|
||||
~GmockWddm() = default;
|
||||
bool virtualFreeWrapper(void *ptr, size_t size, uint32_t flags) {
|
||||
return true;
|
||||
}
|
||||
|
||||
void *virtualAllocWrapper(void *inPtr, size_t size, uint32_t flags, uint32_t type);
|
||||
uintptr_t virtualAllocAddress;
|
||||
MOCK_METHOD5(makeResident, bool(const D3DKMT_HANDLE *handles, uint32_t count, bool cantTrimFurther, uint64_t *numberOfBytesToTrim, size_t totalSize));
|
||||
MOCK_METHOD3(evict, bool(const D3DKMT_HANDLE *handles, uint32_t num, uint64_t &sizeToTrim));
|
||||
MOCK_METHOD1(createAllocationsAndMapGpuVa, NTSTATUS(OsHandleStorage &osHandles));
|
||||
|
||||
NTSTATUS baseCreateAllocationAndMapGpuVa(OsHandleStorage &osHandles) {
|
||||
return Wddm::createAllocationsAndMapGpuVa(osHandles);
|
||||
}
|
||||
};
|
||||
} // namespace NEO
|
||||
Reference in New Issue
Block a user