compute-runtime/shared/test/common/helpers/debug_manager_state_restore.h

40 lines
1.0 KiB
C++

/*
* Copyright (C) 2018-2023 Intel Corporation
*
* SPDX-License-Identifier: MIT
*
*/
#pragma once
#include "shared/source/debug_settings/debug_settings_manager.h"
using namespace NEO;
class DebugManagerStateRestore {
public:
DebugManagerStateRestore() {
debugVarSnapshot = debugManager.flags;
injectFcnSnapshot = debugManager.injectFcn;
}
~DebugManagerStateRestore() {
debugManager.flags = debugVarSnapshot;
debugManager.injectFcn = injectFcnSnapshot;
#undef DECLARE_DEBUG_VARIABLE
#define DECLARE_DEBUG_VARIABLE(dataType, variableName, defaultValue, description) shrink(debugManager.flags.variableName.getRef());
#include "shared/source/debug_settings/release_variables.inl"
#include "debug_variables.inl"
#undef DECLARE_DEBUG_VARIABLE
}
DebugVariables debugVarSnapshot;
void *injectFcnSnapshot = nullptr;
protected:
void shrink(std::string &flag) {
flag.shrink_to_fit();
}
void shrink(int64_t &flag) {}
void shrink(int32_t &flag) {}
void shrink(bool &flag) {}
};