mirror of
https://github.com/intel/compute-runtime.git
synced 2025-06-28 17:58:30 +08:00
75 lines
2.7 KiB
C++
75 lines
2.7 KiB
C++
/*
|
|
* Copyright (C) 2020-2023 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <level_zero/ze_api.h>
|
|
|
|
#include <memory>
|
|
#include <vector>
|
|
|
|
struct _ze_module_handle_t {};
|
|
|
|
namespace NEO {
|
|
struct KernelDescriptor;
|
|
}
|
|
|
|
namespace L0 {
|
|
struct Device;
|
|
struct ModuleBuildLog;
|
|
struct KernelImmutableData;
|
|
|
|
enum class ModuleType {
|
|
builtin,
|
|
user
|
|
};
|
|
|
|
struct Module : _ze_module_handle_t {
|
|
|
|
static Module *create(Device *device, const ze_module_desc_t *desc, ModuleBuildLog *moduleBuildLog, ModuleType type, ze_result_t *result);
|
|
|
|
virtual ~Module() = default;
|
|
|
|
virtual Device *getDevice() const = 0;
|
|
|
|
virtual ze_result_t createKernel(const ze_kernel_desc_t *desc,
|
|
ze_kernel_handle_t *kernelHandle) = 0;
|
|
virtual ze_result_t destroy() = 0;
|
|
virtual ze_result_t getNativeBinary(size_t *pSize, uint8_t *pModuleNativeBinary) = 0;
|
|
virtual ze_result_t getFunctionPointer(const char *pKernelName, void **pfnFunction) = 0;
|
|
virtual ze_result_t getGlobalPointer(const char *pGlobalName, size_t *pSize, void **pPtr) = 0;
|
|
virtual ze_result_t getDebugInfo(size_t *pDebugDataSize, uint8_t *pDebugData) = 0;
|
|
virtual ze_result_t getKernelNames(uint32_t *pCount, const char **pNames) = 0;
|
|
virtual ze_result_t getProperties(ze_module_properties_t *pModuleProperties) = 0;
|
|
virtual ze_result_t performDynamicLink(uint32_t numModules,
|
|
ze_module_handle_t *phModules,
|
|
ze_module_build_log_handle_t *phLinkLog) = 0;
|
|
virtual ze_result_t inspectLinkage(ze_linkage_inspection_ext_desc_t *pInspectDesc,
|
|
uint32_t numModules,
|
|
ze_module_handle_t *phModules,
|
|
ze_module_build_log_handle_t *phLog) = 0;
|
|
|
|
virtual const KernelImmutableData *getKernelImmutableData(const char *kernelName) const = 0;
|
|
virtual const std::vector<std::unique_ptr<KernelImmutableData>> &getKernelImmutableDataVector() const = 0;
|
|
virtual uint32_t getMaxGroupSize(const NEO::KernelDescriptor &kernelDescriptor) const = 0;
|
|
virtual bool shouldAllocatePrivateMemoryPerDispatch() const = 0;
|
|
virtual uint32_t getProfileFlags() const = 0;
|
|
virtual void checkIfPrivateMemoryPerDispatchIsNeeded() = 0;
|
|
|
|
Module() = default;
|
|
Module(const Module &) = delete;
|
|
Module(Module &&) = delete;
|
|
Module &operator=(const Module &) = delete;
|
|
Module &operator=(Module &&) = delete;
|
|
|
|
static Module *fromHandle(ze_module_handle_t handle) { return static_cast<Module *>(handle); }
|
|
|
|
inline ze_module_handle_t toHandle() { return this; }
|
|
};
|
|
|
|
} // namespace L0
|