mirror of
https://github.com/intel/compute-runtime.git
synced 2025-12-30 01:35:20 +08:00
1186c1aae3
Before performing gpu device reset, first all level zero resources and gpu device specific resources have to be cleaned up. Also as after device reset, state of gpu device would be lost. Hence after performing gpu device reset, level zero device have to be reinitialized by querying gpu device again. This change is aimed at reinitializing the level zero resources after gpu device reset, so that user could continue using level zero devices after device reset. Related-To: LOCI-2627 Signed-off-by: Jitendra Sharma <jitendra.sharma@intel.com>
50 lines
1.2 KiB
C++
50 lines
1.2 KiB
C++
/*
|
|
* Copyright (C) 2020-2021 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
#include <level_zero/zes_api.h>
|
|
|
|
#include <map>
|
|
#include <vector>
|
|
|
|
struct _zes_engine_handle_t {
|
|
virtual ~_zes_engine_handle_t() = default;
|
|
};
|
|
|
|
namespace L0 {
|
|
using EngineInstanceSubDeviceId = std::pair<uint32_t, uint32_t>;
|
|
struct OsSysman;
|
|
|
|
class Engine : _zes_engine_handle_t {
|
|
public:
|
|
virtual ze_result_t engineGetProperties(zes_engine_properties_t *pProperties) = 0;
|
|
virtual ze_result_t engineGetActivity(zes_engine_stats_t *pStats) = 0;
|
|
|
|
static Engine *fromHandle(zes_engine_handle_t handle) {
|
|
return static_cast<Engine *>(handle);
|
|
}
|
|
inline zes_engine_handle_t toHandle() { return this; }
|
|
};
|
|
|
|
struct EngineHandleContext {
|
|
EngineHandleContext(OsSysman *pOsSysman);
|
|
MOCKABLE_VIRTUAL ~EngineHandleContext();
|
|
|
|
MOCKABLE_VIRTUAL void init();
|
|
void releaseEngines();
|
|
|
|
ze_result_t engineGet(uint32_t *pCount, zes_engine_handle_t *phEngine);
|
|
|
|
OsSysman *pOsSysman = nullptr;
|
|
std::vector<Engine *> handleList = {};
|
|
|
|
private:
|
|
void createHandle(zes_engine_group_t engineType, uint32_t engineInstance, uint32_t subDeviceId);
|
|
};
|
|
|
|
} // namespace L0
|