6565345316

Committed 18 Oct 2023 06:58PM UTC coverage: 78.209% (+0.04%) from 78.167%

Build # 6565345316

Build Type

Pull #1179

github

Committed by

web-flow

Commit Message

Merge f904a77d8 into 9e6c22490

Pull Request Pull Request #1179: Add sycl event wait overload

Run Details

1075 of 1624 branches covered (0.0%)

Branch coverage included in aggregate %.

23 of 24 new or added lines in 2 files covered. (95.83%)

4 existing lines in 1 file now uncovered.

4682 of 5737 relevant lines covered (81.61%)

0.82 hits per line

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

88.89

/numba_dpex/dpctl_iface/_intrinsic.py

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

from numba import types
from numba.core import cgutils
from numba.core.datamodel import default_manager
from numba.extending import intrinsic, overload_method

import numba_dpex.dpctl_iface.libsyclinterface_bindings as sycl
from numba_dpex.core import types as dpex_types


@intrinsic
def sycl_event_wait(typingctx, ty_event):
    # check for accepted types
    if not isinstance(ty_event, dpex_types.DpctlSyclEvent):
        return

    result_type = types.void
    sig = result_type(ty_event)

    # defines the custom code generation
    def codegen(context, builder, signature, args):
        event_struct_proxy = cgutils.create_struct_proxy(ty_event)(
            context, builder
        )
        event_struct_ptr = event_struct_proxy._getpointer()

        event_struct = builder.load(event_struct_ptr)
        sycl_event_dm = default_manager.lookup(ty_event)
        event_ref = builder.extract_value(
            event_struct,
            sycl_event_dm.get_field_position("event_ref"),
        )

        sycl.dpctl_event_wait(builder, event_ref)

    return sig, codegen


@overload_method(dpex_types.DpctlSyclEvent, "wait")
def ol_dpctl_sycl_event_wait(
    event,
):
    """Implementation of an overload to support dpctl.SyclEvent() inside
    a dpjit function.
    """
    return lambda event: sycl_event_wait(event)

1	# SPDX-FileCopyrightText: 2020 - 2023 Intel Corporation
2	#
3	# SPDX-License-Identifier: Apache-2.0
4
5	from numba import types	1✔
6	from numba.core import cgutils	1✔
7	from numba.core.datamodel import default_manager	1✔
8	from numba.extending import intrinsic, overload_method	1✔
9
10	import numba_dpex.dpctl_iface.libsyclinterface_bindings as sycl	1✔
11	from numba_dpex.core import types as dpex_types	1✔
12
13
14	@intrinsic	1✔
15	def sycl_event_wait(typingctx, ty_event):	1✔
16	# check for accepted types
17	if not isinstance(ty_event, dpex_types.DpctlSyclEvent):	1!
NEW 18	return	×
19
20	result_type = types.void	1✔
21	sig = result_type(ty_event)	1✔
22
23	# defines the custom code generation
24	def codegen(context, builder, signature, args):	1✔
25	event_struct_proxy = cgutils.create_struct_proxy(ty_event)(	1✔
26	context, builder
27	)
28	event_struct_ptr = event_struct_proxy._getpointer()	1✔
29
30	event_struct = builder.load(event_struct_ptr)	1✔
31	sycl_event_dm = default_manager.lookup(ty_event)	1✔
32	event_ref = builder.extract_value(	1✔
33	event_struct,
34	sycl_event_dm.get_field_position("event_ref"),
35	)
36
37	sycl.dpctl_event_wait(builder, event_ref)	1✔
38
39	return sig, codegen	1✔
40
41
42	@overload_method(dpex_types.DpctlSyclEvent, "wait")	1✔
43	def ol_dpctl_sycl_event_wait(	1✔
44	event,
45	):
46	"""Implementation of an overload to support dpctl.SyclEvent() inside
47	a dpjit function.
48	"""
49	return lambda event: sycl_event_wait(event)	1!

IntelPython / numba-dpex / 6565345316

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