compute-runtime/opencl/test/unit_test/global_environment.h

71 lines
1.8 KiB
C++

/*
* Copyright (C) 2018-2021 Intel Corporation
*
* SPDX-License-Identifier: MIT
*
*/
#pragma once
#include "shared/source/helpers/hw_info.h"
#include "shared/source/os_interface/os_library.h"
#include "shared/test/common/mocks/mock_compilers.h"
#include "gtest/gtest.h"
using namespace NEO;
OsLibrary *setAdapterInfo(const PLATFORM *platform, const GT_SYSTEM_INFO *gtSystemInfo, uint64_t gpuAddressSpace);
class TestEnvironment : public ::testing::Environment {
public:
TestEnvironment();
void SetUp() override;
void TearDown() override;
virtual void fclPushDebugVars(
MockCompilerDebugVars &newDebugVars);
virtual void fclPopDebugVars();
virtual void igcPushDebugVars(
MockCompilerDebugVars &newDebugVars);
virtual void igcPopDebugVars();
virtual void setDefaultDebugVars(
MockCompilerDebugVars &fclDefaults,
MockCompilerDebugVars &igcDefaults,
HardwareInfo &hwInfo);
virtual void setMockFileNames(
std::string &fclMockFile,
std::string &igcMockFile);
virtual std::string &fclGetMockFile();
virtual std::string &igcGetMockFile();
protected:
OsLibrary *libraryFrontEnd = nullptr;
OsLibrary *libraryIGC = nullptr;
OsLibrary *libraryOS = nullptr;
std::vector<MockCompilerDebugVars> igcDebugVarStack;
std::vector<MockCompilerDebugVars> fclDebugVarStack;
void (*igcSetDebugVarsFPtr)(MockCompilerDebugVars &debugVars);
void (*fclSetDebugVarsFptr)(MockCompilerDebugVars &debugVars);
MockCompilerDebugVars fclDefaultDebugVars{};
MockCompilerDebugVars igcDefaultDebugVars{};
HardwareInfo hwInfoDefaultDebugVars{};
std::string fclMockFile{};
std::string igcMockFile{};
MockCompilerEnableGuard mockCompilerGuard{};
};
extern TestEnvironment *gEnvironment;