6565981927

Committed 18 Oct 2023 08:04PM UTC coverage: 74.191% (-4.0%) from 78.167%

Build # 6565981927

Build Type

Pull #1178

github

Committed by

web-flow

Commit Message

Merge e8a1a96d4 into 9e6c22490

Pull Request Pull Request #1178: An experimental kernel dispatcher for numba_dpex.kernel decorator

Run Details

1074 of 1696 branches covered (0.0%)

Branch coverage included in aggregate %.

68 of 386 new or added lines in 8 files covered. (17.62%)

10 existing lines in 1 file now uncovered.

4684 of 6065 relevant lines covered (77.23%)

0.77 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

55.56

/numba_dpex/dpctl_iface/libsyclinterface_bindings.py

# SPDX-FileCopyrightText: 2023 Intel Corporation
#
# SPDX-License-Identifier: Apache-2.0

"""llvmlite function call generators for dpctl's libsyclinterface library"""

from llvmlite import ir as llvmir
from numba.core import cgutils


def _build_dpctl_function(llvm_module, return_ty, arg_list, func_name):
    """
    _build_dpctl_function(llvm_module, return_ty, arg_list, func_name)
    Inserts an LLVM function of the specified signature and name into an
    LLVM module.

    Args:
        llvm_module: The LLVM IR Module into which the function will be inserted
        return_ty: An LLVM Value corresponding to the return type of the
                   function.
        arg_list: A list of LLVM Value objects corresponding to the
                  type of the arguments for the function.
        func_name: The name of the function passed as a string.

    Return: A Python object wrapping an LLVM Function.

    """
    func_ty = llvmir.FunctionType(return_ty, arg_list)
    fn = cgutils.get_or_insert_function(llvm_module, func_ty, func_name)
    return fn


# --------------------- DpctlSyclQueue functions ------------------------------#


def dpctl_queue_copy(builder: llvmir.IRBuilder, *args):
    """Inserts LLVM IR to call DPCTLQueue_Copy to create a copy of a
    DpctlSyclQueueRef pointer passed in to the function.
    """
    mod = builder.module
    fn = _build_dpctl_function(
        llvm_module=mod,
        return_ty=cgutils.voidptr_t,
        arg_list=[cgutils.voidptr_t],
        func_name="DPCTLQueue_Copy",
    )

    ret = builder.call(fn, args)

    return ret


def dpctl_queue_memcpy(builder: llvmir.IRBuilder, *args):
    """Inserts LLVM IR to call DPCTLQueue_Memcpy."""
    mod = builder.module
    fn = _build_dpctl_function(
        llvm_module=mod,
        return_ty=cgutils.voidptr_t,
        arg_list=[
            cgutils.voidptr_t,
            cgutils.voidptr_t,
            cgutils.voidptr_t,
            cgutils.intp_t,
        ],
        func_name="DPCTLQueue_Memcpy",
    )
    ret = builder.call(fn, args)

    return ret


def dpctl_queue_delete(builder: llvmir.IRBuilder, *args):
    """Inserts LLVM IR to call DPCTLQueue_Delete."""
    mod = builder.module
    fn = _build_dpctl_function(
        llvm_module=mod,
        return_ty=llvmir.types.VoidType(),
        arg_list=[cgutils.voidptr_t],
        func_name="DPCTLQueue_Delete",
    )
    ret = builder.call(fn, args)

    return ret


def dpctl_queue_get_context(builder: llvmir.IRBuilder, *args):
    """Inserts LLVM IR to call DPCTLQueue_GetContext."""
    mod = builder.module
    fn = _build_dpctl_function(
        llvm_module=mod,
        return_ty=cgutils.voidptr_t,
        arg_list=[cgutils.voidptr_t],
        func_name="DPCTLQueue_GetContext",
    )
    ret = builder.call(fn, args)

    return ret


def dpctl_queue_get_device(builder: llvmir.IRBuilder, *args):
    """Inserts LLVM IR to call DPCTLQueue_GetDevice."""
    mod = builder.module
    fn = _build_dpctl_function(
        llvm_module=mod,
        return_ty=cgutils.voidptr_t,
        arg_list=[cgutils.voidptr_t],
        func_name="DPCTLQueue_GetDevice",
    )
    ret = builder.call(fn, args)

    return ret


# --------------------- DpctlSyclDevice functions -----------------------------#


def dpctl_device_delete(builder: llvmir.IRBuilder, *args):
    """Inserts LLVM IR to call DPCTLDevice_Delete."""
    mod = builder.module
    fn = _build_dpctl_function(
        llvm_module=mod,
        return_ty=llvmir.types.VoidType(),
        arg_list=[cgutils.voidptr_t],
        func_name="DPCTLDevice_Delete",
    )
    ret = builder.call(fn, args)

    return ret


# --------------------- DpctlSyclKernelBundle functions -----------------------#


def dpctl_kernel_bundle_create_from_spirv(builder: llvmir.IRBuilder, *args):
    """Inserts LLVM IR to call DPCTLKernelBundle_CreateFromSpirv."""
    mod = builder.module
    fn = _build_dpctl_function(
        llvm_module=mod,
        return_ty=cgutils.voidptr_t,
        arg_list=[
            cgutils.voidptr_t,
            cgutils.voidptr_t,
            cgutils.voidptr_t,
            llvmir.IntType(64),
            cgutils.voidptr_t,
        ],
        func_name="DPCTLKernelBundle_CreateFromSpirv",
    )
    ret = builder.call(fn, args)

    return ret


def dpctl_kernel_bundle_delete(builder: llvmir.IRBuilder, *args):
    """Inserts LLVM IR to call DPCTLKernelBundle_Delete."""
    mod = builder.module
    fn = _build_dpctl_function(
        llvm_module=mod,
        return_ty=llvmir.types.VoidType(),
        arg_list=[cgutils.voidptr_t],
        func_name="DPCTLKernelBundle_Delete",
    )
    ret = builder.call(fn, args)

    return ret


def dpctl_kernel_bundle_get_kernel(builder: llvmir.IRBuilder, *args):
    """Inserts LLVM IR to call DPCTLKernelBundle_GetKernel."""
    mod = builder.module
    fn = _build_dpctl_function(
        llvm_module=mod,
        return_ty=cgutils.voidptr_t,
        arg_list=[cgutils.voidptr_t, cgutils.voidptr_t],
        func_name="DPCTLKernelBundle_GetKernel",
    )
    ret = builder.call(fn, args)

    return ret


# --------------------- DpctlSyclKernel functions -----------------------------#


def dpctl_kernel_delete(builder: llvmir.IRBuilder, *args):
    """Inserts LLVM IR to call DPCTLKernel_Delete."""
    mod = builder.module
    fn = _build_dpctl_function(
        llvm_module=mod,
        return_ty=llvmir.types.VoidType(),
        arg_list=[cgutils.voidptr_t],
        func_name="DPCTLKernel_Delete",
    )
    ret = builder.call(fn, args)

    return ret


# --------------------- DpctlSyclContext functions ----------------------------#


def dpctl_context_delete(builder: llvmir.IRBuilder, *args):
    """Inserts LLVM IR to call DPCTLContext_Delete."""
    mod = builder.module
    fn = _build_dpctl_function(
        llvm_module=mod,
        return_ty=llvmir.types.VoidType(),
        arg_list=[cgutils.voidptr_t],
        func_name="DPCTLContext_Delete",
    )
    ret = builder.call(fn, args)

    return ret


# --------------------- DpctlSyclEvent functions ------------------------------#


def dpctl_event_wait(builder: llvmir.IRBuilder, *args):
    """Inserts LLVM IR to call DPCTLEvent_Wait."""
    mod = builder.module
    fn = _build_dpctl_function(
        llvm_module=mod,
        return_ty=llvmir.types.VoidType(),
        arg_list=[cgutils.voidptr_t],
        func_name="DPCTLEvent_Wait",
    )
    ret = builder.call(fn, args)

    return ret


def dpctl_event_delete(builder: llvmir.IRBuilder, *args):
    """Inserts LLVM IR to call DPCTLEvent_Delete."""
    mod = builder.module
    fn = _build_dpctl_function(
        llvm_module=mod,
        return_ty=llvmir.types.VoidType(),
        arg_list=[cgutils.voidptr_t],
        func_name="DPCTLEvent_Delete",
    )
    ret = builder.call(fn, args)

    return ret

1	# SPDX-FileCopyrightText: 2023 Intel Corporation
2	#
3	# SPDX-License-Identifier: Apache-2.0
4
5	"""llvmlite function call generators for dpctl's libsyclinterface library"""	1✔
6
7	from llvmlite import ir as llvmir	1✔
8	from numba.core import cgutils	1✔
9
10
11	def _build_dpctl_function(llvm_module, return_ty, arg_list, func_name):	1✔
12	"""
13	_build_dpctl_function(llvm_module, return_ty, arg_list, func_name)
14	Inserts an LLVM function of the specified signature and name into an
15	LLVM module.
16
17	Args:
18	llvm_module: The LLVM IR Module into which the function will be inserted
19	return_ty: An LLVM Value corresponding to the return type of the
20	function.
21	arg_list: A list of LLVM Value objects corresponding to the
22	type of the arguments for the function.
23	func_name: The name of the function passed as a string.
24
25	Return: A Python object wrapping an LLVM Function.
26
27	"""
28	func_ty = llvmir.FunctionType(return_ty, arg_list)	1✔
29	fn = cgutils.get_or_insert_function(llvm_module, func_ty, func_name)	1✔
30	return fn	1✔
31
32
33	# --------------------- DpctlSyclQueue functions ------------------------------#
34
35
36	def dpctl_queue_copy(builder: llvmir.IRBuilder, *args):	1✔
37	"""Inserts LLVM IR to call DPCTLQueue_Copy to create a copy of a
38	DpctlSyclQueueRef pointer passed in to the function.
39	"""
40	mod = builder.module	1✔
41	fn = _build_dpctl_function(	1✔
42	llvm_module=mod,
43	return_ty=cgutils.voidptr_t,
44	arg_list=[cgutils.voidptr_t],
45	func_name="DPCTLQueue_Copy",
46	)
47
48	ret = builder.call(fn, args)	1✔
49
50	return ret	1✔
51
52
53	def dpctl_queue_memcpy(builder: llvmir.IRBuilder, *args):	1✔
54	"""Inserts LLVM IR to call DPCTLQueue_Memcpy."""
55	mod = builder.module	1✔
56	fn = _build_dpctl_function(	1✔
57	llvm_module=mod,
58	return_ty=cgutils.voidptr_t,
59	arg_list=[
60	cgutils.voidptr_t,
61	cgutils.voidptr_t,
62	cgutils.voidptr_t,
63	cgutils.intp_t,
64	],
65	func_name="DPCTLQueue_Memcpy",
66	)
67	ret = builder.call(fn, args)	1✔
68
69	return ret	1✔
70
71
72	def dpctl_queue_delete(builder: llvmir.IRBuilder, *args):	1✔
73	"""Inserts LLVM IR to call DPCTLQueue_Delete."""
74	mod = builder.module	1✔
75	fn = _build_dpctl_function(	1✔
76	llvm_module=mod,
77	return_ty=llvmir.types.VoidType(),
78	arg_list=[cgutils.voidptr_t],
79	func_name="DPCTLQueue_Delete",
80	)
81	ret = builder.call(fn, args)	1✔
82
83	return ret	1✔
84
85
86	def dpctl_queue_get_context(builder: llvmir.IRBuilder, *args):	1✔
87	"""Inserts LLVM IR to call DPCTLQueue_GetContext."""
UNCOV 88	mod = builder.module	×
UNCOV 89	fn = _build_dpctl_function(	×
90	llvm_module=mod,
91	return_ty=cgutils.voidptr_t,
92	arg_list=[cgutils.voidptr_t],
93	func_name="DPCTLQueue_GetContext",
94	)
UNCOV 95	ret = builder.call(fn, args)	×
96
UNCOV 97	return ret	×
98
99
100	def dpctl_queue_get_device(builder: llvmir.IRBuilder, *args):	1✔
101	"""Inserts LLVM IR to call DPCTLQueue_GetDevice."""
UNCOV 102	mod = builder.module	×
UNCOV 103	fn = _build_dpctl_function(	×
104	llvm_module=mod,
105	return_ty=cgutils.voidptr_t,
106	arg_list=[cgutils.voidptr_t],
107	func_name="DPCTLQueue_GetDevice",
108	)
UNCOV 109	ret = builder.call(fn, args)	×
110
UNCOV 111	return ret	×
112
113
114	# --------------------- DpctlSyclDevice functions -----------------------------#
115
116
117	def dpctl_device_delete(builder: llvmir.IRBuilder, *args):	1✔
118	"""Inserts LLVM IR to call DPCTLDevice_Delete."""
NEW 119	mod = builder.module	×
NEW 120	fn = _build_dpctl_function(	×
121	llvm_module=mod,
122	return_ty=llvmir.types.VoidType(),
123	arg_list=[cgutils.voidptr_t],
124	func_name="DPCTLDevice_Delete",
125	)
NEW 126	ret = builder.call(fn, args)	×
127
NEW 128	return ret	×
129
130
131	# --------------------- DpctlSyclKernelBundle functions -----------------------#
132
133
134	def dpctl_kernel_bundle_create_from_spirv(builder: llvmir.IRBuilder, *args):	1✔
135	"""Inserts LLVM IR to call DPCTLKernelBundle_CreateFromSpirv."""
UNCOV 136	mod = builder.module	×
UNCOV 137	fn = _build_dpctl_function(	×
138	llvm_module=mod,
139	return_ty=cgutils.voidptr_t,
140	arg_list=[
141	cgutils.voidptr_t,
142	cgutils.voidptr_t,
143	cgutils.voidptr_t,
144	llvmir.IntType(64),
145	cgutils.voidptr_t,
146	],
147	func_name="DPCTLKernelBundle_CreateFromSpirv",
148	)
NEW 149	ret = builder.call(fn, args)	×
150
NEW 151	return ret	×
152
153
154	def dpctl_kernel_bundle_delete(builder: llvmir.IRBuilder, *args):	1✔
155	"""Inserts LLVM IR to call DPCTLKernelBundle_Delete."""
NEW 156	mod = builder.module	×
NEW 157	fn = _build_dpctl_function(	×
158	llvm_module=mod,
159	return_ty=llvmir.types.VoidType(),
160	arg_list=[cgutils.voidptr_t],
161	func_name="DPCTLKernelBundle_Delete",
162	)
NEW 163	ret = builder.call(fn, args)	×
164
NEW 165	return ret	×
166
167
168	def dpctl_kernel_bundle_get_kernel(builder: llvmir.IRBuilder, *args):	1✔
169	"""Inserts LLVM IR to call DPCTLKernelBundle_GetKernel."""
NEW 170	mod = builder.module	×
NEW 171	fn = _build_dpctl_function(	×
172	llvm_module=mod,
173	return_ty=cgutils.voidptr_t,
174	arg_list=[cgutils.voidptr_t, cgutils.voidptr_t],
175	func_name="DPCTLKernelBundle_GetKernel",
176	)
NEW 177	ret = builder.call(fn, args)	×
178
NEW 179	return ret	×
180
181
182	# --------------------- DpctlSyclKernel functions -----------------------------#
183
184
185	def dpctl_kernel_delete(builder: llvmir.IRBuilder, *args):	1✔
186	"""Inserts LLVM IR to call DPCTLKernel_Delete."""
NEW 187	mod = builder.module	×
NEW 188	fn = _build_dpctl_function(	×
189	llvm_module=mod,
190	return_ty=llvmir.types.VoidType(),
191	arg_list=[cgutils.voidptr_t],
192	func_name="DPCTLKernel_Delete",
193	)
NEW 194	ret = builder.call(fn, args)	×
195
NEW 196	return ret	×
197
198
199	# --------------------- DpctlSyclContext functions ----------------------------#
200
201
202	def dpctl_context_delete(builder: llvmir.IRBuilder, *args):	1✔
203	"""Inserts LLVM IR to call DPCTLContext_Delete."""
NEW 204	mod = builder.module	×
NEW 205	fn = _build_dpctl_function(	×
206	llvm_module=mod,
207	return_ty=llvmir.types.VoidType(),
208	arg_list=[cgutils.voidptr_t],
209	func_name="DPCTLContext_Delete",
210	)
NEW 211	ret = builder.call(fn, args)	×
212
NEW 213	return ret	×
214
215
216	# --------------------- DpctlSyclEvent functions ------------------------------#
217
218
219	def dpctl_event_wait(builder: llvmir.IRBuilder, *args):	1✔
220	"""Inserts LLVM IR to call DPCTLEvent_Wait."""
221	mod = builder.module	1✔
222	fn = _build_dpctl_function(	1✔
223	llvm_module=mod,
224	return_ty=llvmir.types.VoidType(),
225	arg_list=[cgutils.voidptr_t],
226	func_name="DPCTLEvent_Wait",
227	)
228	ret = builder.call(fn, args)	1✔
229
230	return ret	1✔
231
232
233	def dpctl_event_delete(builder: llvmir.IRBuilder, *args):	1✔
234	"""Inserts LLVM IR to call DPCTLEvent_Delete."""
235	mod = builder.module	1✔
236	fn = _build_dpctl_function(	1✔
237	llvm_module=mod,
238	return_ty=llvmir.types.VoidType(),
239	arg_list=[cgutils.voidptr_t],
240	func_name="DPCTLEvent_Delete",
241	)
242	ret = builder.call(fn, args)	1✔
243
244	return ret	1✔

IntelPython / numba-dpex / 6565981927

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous