compute-runtime/shared/test/common/libult/linux/drm_query_mock.h

47 lines
1.3 KiB
C++

/*
* Copyright (C) 2022-2023 Intel Corporation
*
* SPDX-License-Identifier: MIT
*
*/
#pragma once
#include "shared/test/common/helpers/default_hw_info.h"
#include "shared/test/common/libult/linux/drm_mock.h"
#include "shared/test/common/libult/linux/drm_mock_prelim_context.h"
using namespace NEO;
class DrmQueryMock : public DrmMock {
public:
using Drm::rootDeviceEnvironment;
DrmQueryMock(RootDeviceEnvironment &rootDeviceEnvironment);
DrmMockPrelimContext context{
nullptr,
rootDeviceEnvironment,
getCacheInfo(),
failRetTopology,
supportedCopyEnginesMask,
contextDebugSupported,
};
void getPrelimEuDebug(int &prelimEuDebug) override {
prelimEuDebug = prelimEuDebugValue;
}
int prelimEuDebugValue = 0;
static constexpr uint32_t maxEngineCount{9};
ContextEnginesLoadBalance<maxEngineCount> receivedContextEnginesLoadBalance{};
ContextParamEngines<1 + maxEngineCount> receivedContextParamEngines{};
BcsInfoMask supportedCopyEnginesMask = 1;
uint32_t i915QuerySuccessCount = std::numeric_limits<uint32_t>::max();
int storedRetValForSetParamEngines{0};
int handleRemainingRequests(DrmIoctl request, void *arg) override;
virtual bool handleQueryItem(void *queryItem);
};