compute-runtime/shared/test/common/fixtures/device_fixture.h

44 lines
1021 B
C++

/*
* Copyright (C) 2021-2024 Intel Corporation
*
* SPDX-License-Identifier: MIT
*
*/
#pragma once
#include "shared/source/command_stream/task_count_helper.h"
#include "shared/source/helpers/hw_info.h"
#include "shared/test/common/helpers/debug_manager_state_restore.h"
namespace NEO {
class MockDevice;
class ReleaseHelper;
struct DeviceFixture {
void setUp();
void setUpImpl(const NEO::HardwareInfo *hardwareInfo);
void tearDown();
MockDevice *createWithUsDeviceIdRevId(unsigned short usDeviceId, unsigned short usRevId);
MockDevice *pDevice = nullptr;
volatile TagAddressType *pTagMemory = nullptr;
HardwareInfo hardwareInfo = {};
PLATFORM platformHelper = {};
const uint32_t rootDeviceIndex = 0u;
template <typename HelperType>
HelperType &getHelper() const;
const ReleaseHelper *getReleaseHelper();
};
struct DeviceWithoutSipFixture : DeviceFixture {
void setUp();
void tearDown();
DebugManagerStateRestore dbgRestorer;
};
} // namespace NEO