sysman:add reinitialization code to diagnostics

Signed-off-by: T J Vivek Vilvaraj <[email protected]>

Author: vilvarajintel

level_zero/tools/source/sysman/diagnostics/linux/os_diagnostics_imp.cpp

@@ -9,6 +9,8 @@
 
 #include "shared/source/helpers/string.h"
 
+#include "level_zero/core/source/device/device_imp.h"
+
 namespace L0 {
 const std::string LinuxDiagnosticsImp::deviceDir("device");
 
@@ -34,6 +36,10 @@ LinuxDiagnosticsImp::LinuxDiagnosticsImp(OsSysman *pOsSysman, const std::string
     pSysfsAccess = &pLinuxSysmanImp->getSysfsAccess();
     pFsAccess = &pLinuxSysmanImp->getFsAccess();
     pProcfsAccess = &pLinuxSysmanImp->getProcfsAccess();
+    pDevice = pLinuxSysmanImp->getDeviceHandle();
+    auto device = static_cast<DeviceImp *>(pDevice);
+    executionEnvironment = device->getNEODevice()->getExecutionEnvironment();
+    rootDeviceIndex = device->getNEODevice()->getRootDeviceIndex();
 }
 
 std::unique_ptr<OsDiagnostics> OsDiagnostics::create(OsSysman *pOsSysman, const std::string &diagTests, ze_bool_t onSubdevice, uint32_t subdeviceId) {

level_zero/tools/source/sysman/diagnostics/linux/os_diagnostics_imp.h

@@ -33,10 +33,18 @@ class LinuxDiagnosticsImp : public OsDiagnostics, NEO::NonCopyableOrMovableClass
     SysfsAccess *pSysfsAccess = nullptr;
     FsAccess *pFsAccess = nullptr;
     ProcfsAccess *pProcfsAccess = nullptr;
+    Device *pDevice = nullptr;
+    std::string devicePciBdf = "";
+    NEO::ExecutionEnvironment *executionEnvironment = nullptr;
+    uint32_t rootDeviceIndex = 0u;
     decltype(&NEO::SysCalls::open) openFunction = NEO::SysCalls::open;
     decltype(&NEO::SysCalls::close) closeFunction = NEO::SysCalls::close;
     decltype(&NEO::SysCalls::pread) preadFunction = NEO::SysCalls::pread;
     decltype(&NEO::SysCalls::pwrite) pwriteFunction = NEO::SysCalls::pwrite;
+    void releaseSysmanDeviceResources();
+    void releaseDeviceResources();
+    ze_result_t initDevice();
+    void reInitSysmanDeviceResources();
 
   private:
     static const std::string quiescentGpuFile;
@@ -46,4 +54,17 @@ class LinuxDiagnosticsImp : public OsDiagnostics, NEO::NonCopyableOrMovableClass
     static const std::string deviceDir;
 };
 
+class ExecutionEnvironmentRefCountRestore {
+  public:
+    ExecutionEnvironmentRefCountRestore() = delete;
+    ExecutionEnvironmentRefCountRestore(NEO::ExecutionEnvironment *executionEnvironmentRecevied) {
+        executionEnvironment = executionEnvironmentRecevied;
+        executionEnvironment->incRefInternal();
+    }
+    ~ExecutionEnvironmentRefCountRestore() {
+        executionEnvironment->decRefInternal();
+    }
+    NEO::ExecutionEnvironment *executionEnvironment = nullptr;
+};
+
 } // namespace L0

level_zero/tools/test/unit_tests/sources/sysman/diagnostics/linux/mock_zes_sysman_diagnostics.h

@@ -68,6 +68,15 @@ struct Mock<DiagnosticsFwInterface> : public FirmwareUtil {
     ADDMETHOD_NOBASE(fwGetMemoryErrorCount, ze_result_t, ZE_RESULT_SUCCESS, (zes_ras_error_type_t category, uint32_t subDeviceCount, uint32_t subDeviceId, uint64_t &count));
     ADDMETHOD_NOBASE_VOIDRETURN(getDeviceSupportedFwTypes, (std::vector<std::string> & fwTypes));
 };
+struct GlobalOperationsEngineHandleContext : public EngineHandleContext {
+    GlobalOperationsEngineHandleContext(OsSysman *pOsSysman) : EngineHandleContext(pOsSysman) {}
+};
+template <>
+struct Mock<GlobalOperationsEngineHandleContext> : public GlobalOperationsEngineHandleContext {
+    void initMock() {}
+    Mock<GlobalOperationsEngineHandleContext>(OsSysman *pOsSysman) : GlobalOperationsEngineHandleContext(pOsSysman) {}
+    MOCK_METHOD(void, init, (), (override));
+};
 
 class DiagSysfsAccess : public SysfsAccess {};
 template <>
@@ -264,6 +273,7 @@ class PublicLinuxDiagnosticsImp : public L0::LinuxDiagnosticsImp {
   public:
     using LinuxDiagnosticsImp::closeFunction;
     using LinuxDiagnosticsImp::openFunction;
+    using LinuxDiagnosticsImp::pDevice;
     using LinuxDiagnosticsImp::pFsAccess;
     using LinuxDiagnosticsImp::pFwInterface;
     using LinuxDiagnosticsImp::pLinuxSysmanImp;