6499182848

Committed 12 Oct 2023 05:53PM UTC coverage: 76.968% (-0.7%) from 77.682%

Build # 6499182848

Build Type

Pull #1168

github

Committed by

web-flow

Commit Message

Merge 262ed529c into 1ece1358e

Pull Request Pull Request #1168: Cleanups to kernel target

Run Details

1084 of 1652 branches covered (0.0%)

Branch coverage included in aggregate %.

24 of 38 new or added lines in 5 files covered. (63.16%)

25 existing lines in 1 file now uncovered.

4734 of 5907 relevant lines covered (80.14%)

0.8 hits per line

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

59.38

/numba_dpex/core/descriptor.py

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

from functools import cached_property

from numba.core import options, targetconfig, typing
from numba.core.cpu import CPUTargetOptions
from numba.core.descriptors import TargetDescriptor

from .targets.dpjit_target import DPEX_TARGET_NAME, DpexTargetContext
from .targets.kernel_target import (
    DPEX_KERNEL_TARGET_NAME,
    DpexKernelTargetContext,
    DpexKernelTypingContext,
)

_option_mapping = options._mapping


def _inherit_if_not_set(flags, options, name, default=targetconfig._NotSet):
    if name in options:
        setattr(flags, name, options[name])
        return

    cstk = targetconfig.ConfigStack()
    if cstk:
        # inherit
        top = cstk.top()
        if hasattr(top, name):
            setattr(flags, name, getattr(top, name))
            return

    if default is not targetconfig._NotSet:
        setattr(flags, name, default)


class DpexTargetOptions(CPUTargetOptions):
    experimental = _option_mapping("experimental")
    release_gil = _option_mapping("release_gil")

    def finalize(self, flags, options):
        super().finalize(flags, options)
        _inherit_if_not_set(flags, options, "experimental", False)
        _inherit_if_not_set(flags, options, "release_gil", False)


class DpexKernelTarget(TargetDescriptor):
    """
    Implements a target descriptor for numba_dpex.kernel decorated functions.
    """

    options = DpexTargetOptions

    @cached_property
    def _toplevel_target_context(self):
        """Lazily-initialized top-level target context, for all threads."""
        return DpexKernelTargetContext(self.typing_context, self._target_name)

    @cached_property
    def _toplevel_typing_context(self):
        """Lazily-initialized top-level typing context, for all threads."""
        return DpexKernelTypingContext()

    @property
    def target_context(self):
        """
        The target context used by the Dpex compiler pipeline.
        """
        return self._toplevel_target_context

    @property
    def typing_context(self):
        """
        The typing context for used by the Dpex compiler pipeline.
        """
        return self._toplevel_typing_context


class DpexTarget(TargetDescriptor):
    """
    Implements a target descriptor for numba_dpex.dpjit decorated functions.
    """

    options = CPUTargetOptions

    @cached_property
    def _toplevel_target_context(self):
        # Lazily-initialized top-level target context, for all threads
        return DpexTargetContext(self.typing_context, self._target_name)

    @cached_property
    def _toplevel_typing_context(self):
        # Lazily-initialized top-level typing context, for all threads
        return typing.Context()

    @property
    def target_context(self):
        """
        The target context for dpex targets.
        """
        return self._toplevel_target_context

    @property
    def typing_context(self):
        """
        The typing context for dpex targets.
        """
        return self._toplevel_typing_context


# A global instance of the DpexKernelTarget
dpex_kernel_target = DpexKernelTarget(DPEX_KERNEL_TARGET_NAME)

# A global instance of the DpexTarget
dpex_target = DpexTarget(DPEX_TARGET_NAME)

1	# SPDX-FileCopyrightText: 2020 - 2023 Intel Corporation
2	#
3	# SPDX-License-Identifier: Apache-2.0
4
5	from functools import cached_property	1✔
6
7	from numba.core import options, targetconfig, typing	1✔
8	from numba.core.cpu import CPUTargetOptions	1✔
9	from numba.core.descriptors import TargetDescriptor	1✔
10
11	from .targets.dpjit_target import DPEX_TARGET_NAME, DpexTargetContext	1✔
12	from .targets.kernel_target import (	1✔
13	DPEX_KERNEL_TARGET_NAME,
14	DpexKernelTargetContext,
15	DpexKernelTypingContext,
16	)
17
18	_option_mapping = options._mapping	1✔
19
20
21	def _inherit_if_not_set(flags, options, name, default=targetconfig._NotSet):	1✔
NEW 22	if name in options:	×
NEW 23	setattr(flags, name, options[name])	×
NEW 24	return	×
25
NEW 26	cstk = targetconfig.ConfigStack()	×
NEW 27	if cstk:	×
28	# inherit
NEW 29	top = cstk.top()	×
NEW 30	if hasattr(top, name):	×
NEW 31	setattr(flags, name, getattr(top, name))	×
NEW 32	return	×
33
NEW 34	if default is not targetconfig._NotSet:	×
NEW 35	setattr(flags, name, default)	×
36
37
38	class DpexTargetOptions(CPUTargetOptions):	1✔
39	experimental = _option_mapping("experimental")	1✔
40	release_gil = _option_mapping("release_gil")	1✔
41
42	def finalize(self, flags, options):	1✔
NEW 43	super().finalize(flags, options)	×
NEW 44	_inherit_if_not_set(flags, options, "experimental", False)	×
NEW 45	_inherit_if_not_set(flags, options, "release_gil", False)	×
46
47
48	class DpexKernelTarget(TargetDescriptor):	1✔
49	"""
50	Implements a target descriptor for numba_dpex.kernel decorated functions.
51	"""
52
53	options = DpexTargetOptions	1✔
54
55	@cached_property	1✔
56	def _toplevel_target_context(self):	1✔
57	"""Lazily-initialized top-level target context, for all threads."""
58	return DpexKernelTargetContext(self.typing_context, self._target_name)	1✔
59
60	@cached_property	1✔
61	def _toplevel_typing_context(self):	1✔
62	"""Lazily-initialized top-level typing context, for all threads."""
63	return DpexKernelTypingContext()	1✔
64
65	@property	1✔
66	def target_context(self):	1✔
67	"""
68	The target context used by the Dpex compiler pipeline.
69	"""
70	return self._toplevel_target_context	1✔
71
72	@property	1✔
73	def typing_context(self):	1✔
74	"""
75	The typing context for used by the Dpex compiler pipeline.
76	"""
77	return self._toplevel_typing_context	1✔
78
79
80	class DpexTarget(TargetDescriptor):	1✔
81	"""
82	Implements a target descriptor for numba_dpex.dpjit decorated functions.
83	"""
84
85	options = CPUTargetOptions	1✔
86
87	@cached_property	1✔
88	def _toplevel_target_context(self):	1✔
89	# Lazily-initialized top-level target context, for all threads
90	return DpexTargetContext(self.typing_context, self._target_name)	×
91
92	@cached_property	1✔
93	def _toplevel_typing_context(self):	1✔
94	# Lazily-initialized top-level typing context, for all threads
95	return typing.Context()	×
96
97	@property	1✔
98	def target_context(self):	1✔
99	"""
100	The target context for dpex targets.
101	"""
102	return self._toplevel_target_context	×
103
104	@property	1✔
105	def typing_context(self):	1✔
106	"""
107	The typing context for dpex targets.
108	"""
109	return self._toplevel_typing_context	×
110
111
112	# A global instance of the DpexKernelTarget
113	dpex_kernel_target = DpexKernelTarget(DPEX_KERNEL_TARGET_NAME)	1✔
114
115	# A global instance of the DpexTarget
116	dpex_target = DpexTarget(DPEX_TARGET_NAME)	1✔

IntelPython / numba-dpex / 6499182848

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