1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
|
/*
* 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
|
