8394844465

Committed 22 Mar 2024 06:27PM UTC coverage: 79.082% (-0.8%) from 79.874%

Build # 8394844465

Build Type

Pull #1400

github

Committed by

web-flow

Commit Message

Merge 22f434e7a into 611a59711

Pull Request Pull Request #1400: Feature/move experimental decorators to stable

Run Details

1504 of 2209 branches covered (68.09%)

Branch coverage included in aggregate %.

82 of 89 new or added lines in 4 files covered. (92.13%)

87 existing lines in 12 files now uncovered.

6333 of 7701 relevant lines covered (82.24%)

0.82 hits per line

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

84.21

/numba_dpex/core/compiler.py

# SPDX-FileCopyrightText: 2022 - 2024 Intel Corporation
#
# SPDX-License-Identifier: Apache-2.0

from types import FunctionType

from numba.core import compiler, ir
from numba.core import types as numba_types
from numba.core.compiler_lock import global_compiler_lock

from numba_dpex.core import config
from numba_dpex.core.exceptions import (
    KernelHasReturnValueError,
    UnreachableError,
)
from numba_dpex.core.pipelines.kernel_compiler import KernelCompiler


@global_compiler_lock
def compile_with_dpex(
    pyfunc,
    pyfunc_name,
    args,
    return_type,
    target_context,
    typing_context,
    debug=False,
    is_kernel=True,
    extra_compile_flags=None,
):
    """
    Compiles a function using class:`numba_dpex.core.pipelines.KernelCompiler`
    and returns the compiled result.

    Args:
        args: The list of arguments passed to the kernel.
        debug (bool): Optional flag to turn on debug mode compilation.
        extra_compile_flags: Extra flags passed to the compiler.

    Returns:
        cres: Compiled result.

    Raises:
        KernelHasReturnValueError: If the compiled function returns a
        non-void value.
    """
    # First compilation will trigger the initialization of the backend.
    typingctx = typing_context
    targetctx = target_context

    flags = compiler.Flags()
    # Do not compile the function to a binary, just lower to LLVM
    flags.debuginfo = config.DEBUGINFO_DEFAULT
    flags.no_compile = True
    flags.no_cpython_wrapper = True
    flags.no_cfunc_wrapper = True
    flags.nrt = False

    if debug:
        flags.debuginfo = debug

    # Run compilation pipeline
    if isinstance(pyfunc, FunctionType):
        cres = compiler.compile_extra(
            typingctx=typingctx,
            targetctx=targetctx,
            func=pyfunc,
            args=args,
            return_type=return_type,
            flags=flags,
            locals={},
            pipeline_class=KernelCompiler,
        )
    elif isinstance(pyfunc, ir.FunctionIR):
        cres = compiler.compile_ir(
            typingctx=typingctx,
            targetctx=targetctx,
            func_ir=pyfunc,
            args=args,
            return_type=return_type,
            flags=flags,
            locals={},
            pipeline_class=KernelCompiler,
        )
    else:
        raise UnreachableError()

    if (
        is_kernel
        and cres.signature.return_type is not None
        and cres.signature.return_type != numba_types.void
    ):
        raise KernelHasReturnValueError(
            kernel_name=pyfunc_name,
            return_type=cres.signature.return_type,
        )
    # Linking depending libraries
    library = cres.library
    library.finalize()

    return cres

1	# SPDX-FileCopyrightText: 2022 - 2024 Intel Corporation
2	#
3	# SPDX-License-Identifier: Apache-2.0
4
5	from types import FunctionType	1✔
6
7	from numba.core import compiler, ir	1✔
8	from numba.core import types as numba_types	1✔
9	from numba.core.compiler_lock import global_compiler_lock	1✔
10
11	from numba_dpex.core import config	1✔
12	from numba_dpex.core.exceptions import (	1✔
13	KernelHasReturnValueError,
14	UnreachableError,
15	)
16	from numba_dpex.core.pipelines.kernel_compiler import KernelCompiler	1✔
17
18
19	@global_compiler_lock	1✔
20	def compile_with_dpex(	1✔
21	pyfunc,
22	pyfunc_name,
23	args,
24	return_type,
25	target_context,
26	typing_context,
27	debug=False,
28	is_kernel=True,
29	extra_compile_flags=None,
30	):
31	"""
32	Compiles a function using class:`numba_dpex.core.pipelines.KernelCompiler`
33	and returns the compiled result.
34
35	Args:
36	args: The list of arguments passed to the kernel.
37	debug (bool): Optional flag to turn on debug mode compilation.
38	extra_compile_flags: Extra flags passed to the compiler.
39
40	Returns:
41	cres: Compiled result.
42
43	Raises:
44	KernelHasReturnValueError: If the compiled function returns a
45	non-void value.
46	"""
47	# First compilation will trigger the initialization of the backend.
48	typingctx = typing_context	1✔
49	targetctx = target_context	1✔
50
51	flags = compiler.Flags()	1✔
52	# Do not compile the function to a binary, just lower to LLVM
53	flags.debuginfo = config.DEBUGINFO_DEFAULT	1✔
54	flags.no_compile = True	1✔
55	flags.no_cpython_wrapper = True	1✔
56	flags.no_cfunc_wrapper = True	1✔
57	flags.nrt = False	1✔
58
59	if debug:	1!
UNCOV 60	flags.debuginfo = debug	×
61
62	# Run compilation pipeline
63	if isinstance(pyfunc, FunctionType):	1✔
64	cres = compiler.compile_extra(	1✔
65	typingctx=typingctx,
66	targetctx=targetctx,
67	func=pyfunc,
68	args=args,
69	return_type=return_type,
70	flags=flags,
71	locals={},
72	pipeline_class=KernelCompiler,
73	)
74	elif isinstance(pyfunc, ir.FunctionIR):	1!
75	cres = compiler.compile_ir(	1✔
76	typingctx=typingctx,
77	targetctx=targetctx,
78	func_ir=pyfunc,
79	args=args,
80	return_type=return_type,
81	flags=flags,
82	locals={},
83	pipeline_class=KernelCompiler,
84	)
85	else:
86	raise UnreachableError()	×
87
88	if (	1!
89	is_kernel
90	and cres.signature.return_type is not None
91	and cres.signature.return_type != numba_types.void
92	):
93	raise KernelHasReturnValueError(	×
94	kernel_name=pyfunc_name,
95	return_type=cres.signature.return_type,
96	)
97	# Linking depending libraries
98	library = cres.library	1✔
99	library.finalize()	1✔
100
101	return cres	1✔

IntelPython / numba-dpex / 8394844465

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