7047900932

Committed 30 Nov 2023 02:29PM UTC coverage: 79.034% (+0.05%) from 78.988%

Build # 7047900932

Build Type

push

github

Committed by

web-flow

Commit Message

Merge pull request #1219 from IntelPython/feature/async_kernel_submition

Feature/async kernel submition

Run Details

1163 of 1736 branches covered (0.0%)

Branch coverage included in aggregate %.

87 of 92 new or added lines in 6 files covered. (94.57%)

10 existing lines in 2 files now uncovered.

5189 of 6301 relevant lines covered (82.35%)

0.82 hits per line

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

90.8

/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_event_create(builder: llvmir.IRBuilder, *args):
    """Inserts LLVM IR to call DPCTLEvent_Create."""
    mod = builder.module
    fn = _build_dpctl_function(
        llvm_module=mod,
        return_ty=cgutils.voidptr_t,
        arg_list=[],
        func_name="DPCTLEvent_Create",
    )
    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


def dpctl_queue_submit_range(builder: llvmir.IRBuilder, *args):
    """Inserts LLVM IR to call DPCTLQueue_SubmitRange.

    DPCTLSyclEventRef
    DPCTLQueue_SubmitRange(
        const DPCTLSyclKernelRef KRef,
        const DPCTLSyclQueueRef QRef,
        void** Args,
        const DPCTLKernelArgType* ArgTypes,
        size_t NArgs,
        const size_t Range[3],
        size_t NRange,
        const DPCTLSyclEventRef* DepEvents,
        size_t NDepEvents
    );

    """
    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.as_pointer(),
            cgutils.int32_t.as_pointer(),
            llvmir.IntType(64),
            llvmir.IntType(64).as_pointer(),
            llvmir.IntType(64),
            cgutils.voidptr_t,
            llvmir.IntType(64),
        ],
        func_name="DPCTLQueue_SubmitRange",
    )
    ret = builder.call(fn, args)

    return ret


def dpctl_queue_submit_ndrange(builder: llvmir.IRBuilder, *args):
    """Inserts LLVM IR to call DPCTLQueue_SubmitNDRange.

    DPCTLSyclEventRef
    DPCTLQueue_SubmitNDRange(
        const DPCTLSyclKernelRef KRef,
        const DPCTLSyclQueueRef QRef,
        void** Args,
        const DPCTLKernelArgType* ArgTypes,
        size_t NArgs,
        const size_t gRange[3],
        const size_t lRange[3],
        size_t NDims,
        const DPCTLSyclEventRef* DepEvents,
        size_t NDepEvents
    );

    """
    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.as_pointer(),
            cgutils.int32_t.as_pointer(),
            llvmir.IntType(64),
            llvmir.IntType(64).as_pointer(),
            llvmir.IntType(64).as_pointer(),
            llvmir.IntType(64),
            cgutils.voidptr_t,
            llvmir.IntType(64),
        ],
        func_name="DPCTLQueue_SubmitNDRange",
    )
    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.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.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_event_create(builder: llvmir.IRBuilder, *args):	1✔
73	"""Inserts LLVM IR to call DPCTLEvent_Create."""
74	mod = builder.module	1✔
75	fn = _build_dpctl_function(	1✔
76	llvm_module=mod,
77	return_ty=cgutils.voidptr_t,
78	arg_list=[],
79	func_name="DPCTLEvent_Create",
80	)
81	ret = builder.call(fn, args)	1✔
82
83	return ret	1✔
84
85
86	def dpctl_queue_delete(builder: llvmir.IRBuilder, *args):	1✔
87	"""Inserts LLVM IR to call DPCTLQueue_Delete."""
88	mod = builder.module	1✔
89	fn = _build_dpctl_function(	1✔
90	llvm_module=mod,
91	return_ty=llvmir.types.VoidType(),
92	arg_list=[cgutils.voidptr_t],
93	func_name="DPCTLQueue_Delete",
94	)
95	ret = builder.call(fn, args)	1✔
96
97	return ret	1✔
98
99
100	def dpctl_queue_get_context(builder: llvmir.IRBuilder, *args):	1✔
101	"""Inserts LLVM IR to call DPCTLQueue_GetContext."""
102	mod = builder.module	1✔
103	fn = _build_dpctl_function(	1✔
104	llvm_module=mod,
105	return_ty=cgutils.voidptr_t,
106	arg_list=[cgutils.voidptr_t],
107	func_name="DPCTLQueue_GetContext",
108	)
109	ret = builder.call(fn, args)	1✔
110
111	return ret	1✔
112
113
114	def dpctl_queue_get_device(builder: llvmir.IRBuilder, *args):	1✔
115	"""Inserts LLVM IR to call DPCTLQueue_GetDevice."""
116	mod = builder.module	1✔
117	fn = _build_dpctl_function(	1✔
118	llvm_module=mod,
119	return_ty=cgutils.voidptr_t,
120	arg_list=[cgutils.voidptr_t],
121	func_name="DPCTLQueue_GetDevice",
122	)
123	ret = builder.call(fn, args)	1✔
124
125	return ret	1✔
126
127
128	def dpctl_queue_submit_range(builder: llvmir.IRBuilder, *args):	1✔
129	"""Inserts LLVM IR to call DPCTLQueue_SubmitRange.
130
131	DPCTLSyclEventRef
132	DPCTLQueue_SubmitRange(
133	const DPCTLSyclKernelRef KRef,
134	const DPCTLSyclQueueRef QRef,
135	void** Args,
136	const DPCTLKernelArgType* ArgTypes,
137	size_t NArgs,
138	const size_t Range[3],
139	size_t NRange,
140	const DPCTLSyclEventRef* DepEvents,
141	size_t NDepEvents
142	);
143
144	"""
145	mod = builder.module	1✔
146	fn = _build_dpctl_function(	1✔
147	llvm_module=mod,
148	return_ty=cgutils.voidptr_t,
149	arg_list=[
150	cgutils.voidptr_t,
151	cgutils.voidptr_t,
152	cgutils.voidptr_t.as_pointer(),
153	cgutils.int32_t.as_pointer(),
154	llvmir.IntType(64),
155	llvmir.IntType(64).as_pointer(),
156	llvmir.IntType(64),
157	cgutils.voidptr_t,
158	llvmir.IntType(64),
159	],
160	func_name="DPCTLQueue_SubmitRange",
161	)
162	ret = builder.call(fn, args)	1✔
163
164	return ret	1✔
165
166
167	def dpctl_queue_submit_ndrange(builder: llvmir.IRBuilder, *args):	1✔
168	"""Inserts LLVM IR to call DPCTLQueue_SubmitNDRange.
169
170	DPCTLSyclEventRef
171	DPCTLQueue_SubmitNDRange(
172	const DPCTLSyclKernelRef KRef,
173	const DPCTLSyclQueueRef QRef,
174	void** Args,
175	const DPCTLKernelArgType* ArgTypes,
176	size_t NArgs,
177	const size_t gRange[3],
178	const size_t lRange[3],
179	size_t NDims,
180	const DPCTLSyclEventRef* DepEvents,
181	size_t NDepEvents
182	);
183
184	"""
185	mod = builder.module	1✔
186	fn = _build_dpctl_function(	1✔
187	llvm_module=mod,
188	return_ty=cgutils.voidptr_t,
189	arg_list=[
190	cgutils.voidptr_t,
191	cgutils.voidptr_t,
192	cgutils.voidptr_t.as_pointer(),
193	cgutils.int32_t.as_pointer(),
194	llvmir.IntType(64),
195	llvmir.IntType(64).as_pointer(),
196	llvmir.IntType(64).as_pointer(),
197	llvmir.IntType(64),
198	cgutils.voidptr_t,
199	llvmir.IntType(64),
200	],
201	func_name="DPCTLQueue_SubmitNDRange",
202	)
203	ret = builder.call(fn, args)	1✔
204
205	return ret	1✔
206
207
208	# --------------------- DpctlSyclDevice functions -----------------------------#
209
210
211	def dpctl_device_delete(builder: llvmir.IRBuilder, *args):	1✔
212	"""Inserts LLVM IR to call DPCTLDevice_Delete."""
213	mod = builder.module	1✔
214	fn = _build_dpctl_function(	1✔
215	llvm_module=mod,
216	return_ty=llvmir.types.VoidType(),
217	arg_list=[cgutils.voidptr_t],
218	func_name="DPCTLDevice_Delete",
219	)
220	ret = builder.call(fn, args)	1✔
221
222	return ret	1✔
223
224
225	# --------------------- DpctlSyclKernelBundle functions -----------------------#
226
227
228	def dpctl_kernel_bundle_create_from_spirv(builder: llvmir.IRBuilder, *args):	1✔
229	"""Inserts LLVM IR to call DPCTLKernelBundle_CreateFromSpirv."""
230	mod = builder.module	1✔
231	fn = _build_dpctl_function(	1✔
232	llvm_module=mod,
233	return_ty=cgutils.voidptr_t,
234	arg_list=[
235	cgutils.voidptr_t,
236	cgutils.voidptr_t,
237	cgutils.voidptr_t,
238	llvmir.IntType(64),
239	cgutils.voidptr_t,
240	],
241	func_name="DPCTLKernelBundle_CreateFromSpirv",
242	)
243	ret = builder.call(fn, args)	1✔
244
245	return ret	1✔
246
247
248	def dpctl_kernel_bundle_delete(builder: llvmir.IRBuilder, *args):	1✔
249	"""Inserts LLVM IR to call DPCTLKernelBundle_Delete."""
UNCOV 250	mod = builder.module	×
UNCOV 251	fn = _build_dpctl_function(	×
252	llvm_module=mod,
253	return_ty=llvmir.types.VoidType(),
254	arg_list=[cgutils.voidptr_t],
255	func_name="DPCTLKernelBundle_Delete",
256	)
UNCOV 257	ret = builder.call(fn, args)	×
258
UNCOV 259	return ret	×
260
261
262	def dpctl_kernel_bundle_get_kernel(builder: llvmir.IRBuilder, *args):	1✔
263	"""Inserts LLVM IR to call DPCTLKernelBundle_GetKernel."""
264	mod = builder.module	1✔
265	fn = _build_dpctl_function(	1✔
266	llvm_module=mod,
267	return_ty=cgutils.voidptr_t,
268	arg_list=[cgutils.voidptr_t, cgutils.voidptr_t],
269	func_name="DPCTLKernelBundle_GetKernel",
270	)
271	ret = builder.call(fn, args)	1✔
272
273	return ret	1✔
274
275
276	# --------------------- DpctlSyclKernel functions -----------------------------#
277
278
279	def dpctl_kernel_delete(builder: llvmir.IRBuilder, *args):	1✔
280	"""Inserts LLVM IR to call DPCTLKernel_Delete."""
UNCOV 281	mod = builder.module	×
UNCOV 282	fn = _build_dpctl_function(	×
283	llvm_module=mod,
284	return_ty=llvmir.types.VoidType(),
285	arg_list=[cgutils.voidptr_t],
286	func_name="DPCTLKernel_Delete",
287	)
UNCOV 288	ret = builder.call(fn, args)	×
289
UNCOV 290	return ret	×
291
292
293	# --------------------- DpctlSyclContext functions ----------------------------#
294
295
296	def dpctl_context_delete(builder: llvmir.IRBuilder, *args):	1✔
297	"""Inserts LLVM IR to call DPCTLContext_Delete."""
298	mod = builder.module	1✔
299	fn = _build_dpctl_function(	1✔
300	llvm_module=mod,
301	return_ty=llvmir.types.VoidType(),
302	arg_list=[cgutils.voidptr_t],
303	func_name="DPCTLContext_Delete",
304	)
305	ret = builder.call(fn, args)	1✔
306
307	return ret	1✔
308
309
310	# --------------------- DpctlSyclEvent functions ------------------------------#
311
312
313	def dpctl_event_wait(builder: llvmir.IRBuilder, *args):	1✔
314	"""Inserts LLVM IR to call DPCTLEvent_Wait."""
315	mod = builder.module	1✔
316	fn = _build_dpctl_function(	1✔
317	llvm_module=mod,
318	return_ty=llvmir.VoidType(),
319	arg_list=[cgutils.voidptr_t],
320	func_name="DPCTLEvent_Wait",
321	)
322	ret = builder.call(fn, args)	1✔
323
324	return ret	1✔
325
326
327	def dpctl_event_delete(builder: llvmir.IRBuilder, *args):	1✔
328	"""Inserts LLVM IR to call DPCTLEvent_Delete."""
329	mod = builder.module	1✔
330	fn = _build_dpctl_function(	1✔
331	llvm_module=mod,
332	return_ty=llvmir.VoidType(),
333	arg_list=[cgutils.voidptr_t],
334	func_name="DPCTLEvent_Delete",
335	)
336	ret = builder.call(fn, args)	1✔
337
338	return ret	1✔

IntelPython / numba-dpex / 7047900932

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