diff --git a/level_zero/tools/source/debug/debug_session.h b/level_zero/tools/source/debug/debug_session.h index 46a80f3a00..eac5e0dfe0 100644 --- a/level_zero/tools/source/debug/debug_session.h +++ b/level_zero/tools/source/debug/debug_session.h @@ -36,11 +36,20 @@ struct DebugSession : _zet_debug_session_handle_t { virtual ze_result_t writeRegisters(ze_device_thread_t thread, zet_debug_regset_type_t type, uint32_t start, uint32_t count, void *pRegisterValues) = 0; Device *getConnectedDevice() { return connectedDevice; } - virtual void startAsyncThread() = 0; protected: DebugSession(const zet_debug_config_t &config, Device *device) : connectedDevice(device){}; + virtual void startAsyncThread() = 0; + Device *connectedDevice = nullptr; }; +struct RootDebugSession : DebugSession { + virtual ~RootDebugSession() = default; + RootDebugSession() = delete; + + protected: + RootDebugSession(const zet_debug_config_t &config, Device *device) : DebugSession(config, device){}; +}; + } // namespace L0 \ No newline at end of file diff --git a/level_zero/tools/test/unit_tests/sources/debug/mock_debug_session.h b/level_zero/tools/test/unit_tests/sources/debug/mock_debug_session.h index e3c894d3fa..e3f430cb35 100644 --- a/level_zero/tools/test/unit_tests/sources/debug/mock_debug_session.h +++ b/level_zero/tools/test/unit_tests/sources/debug/mock_debug_session.h @@ -23,8 +23,8 @@ class OsInterfaceWithDebugAttach : public NEO::OSInterface { bool debugAttachAvailable = true; }; -struct DebugSessionMock : public L0::DebugSession { - DebugSessionMock(const zet_debug_config_t &config, L0::Device *device) : DebugSession(config, device), config(config){}; +struct DebugSessionMock : public L0::RootDebugSession { + DebugSessionMock(const zet_debug_config_t &config, L0::Device *device) : RootDebugSession(config, device), config(config){}; bool closeConnection() override { return true; } ze_result_t initialize() override { if (config.pid == 0) {