8406344066

Committed 24 Mar 2024 02:37AM UTC coverage: 80.794% (+1.7%) from 79.059%

Build # 8406344066

Build Type

Pull #1404

github

Committed by

web-flow

Commit Message

Merge 33740f9d7 into b1865287d

Pull Request Pull Request #1404: Remove old files not used anymore.

Run Details

1413 of 2013 branches covered (70.19%)

Branch coverage included in aggregate %.

6 of 6 new or added lines in 3 files covered. (100.0%)

18 existing lines in 3 files now uncovered.

5953 of 7104 relevant lines covered (83.8%)

0.84 hits per line

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

0.0

/numba_dpex/core/kernel_interface/utils.py

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

from numba_dpex.core.exceptions import ExecutionQueueInferenceError
from numba_dpex.core.types import USMNdArray


def chk_compute_follows_data_compliance(usm_array_arglist):
    """Check if all the usm ndarray's have the same device.

    Extracts the device filter string from the Numba inferred USMNdArray
    type. Check if the devices corresponding to the filter string are
    equivalent and return a ``dpctl.SyclDevice`` object corresponding to the
    common filter string.

    If an exception occurred in creating a ``dpctl.SyclDevice``, or the
    devices are not equivalent then returns None.

    Args:
        usm_array_arglist : A list of usm_ndarray types specified as
        arguments to the kernel.

    Returns:
        A ``dpctl.SyclDevice`` object if all USMNdArray have same device, or
        else None is returned.
    """

    queue = None

    for usm_array in usm_array_arglist:
        _queue = usm_array.queue
        if not queue:
            queue = _queue
        else:
            if _queue != queue:
                return None

    return queue


def determine_kernel_launch_queue(args, argtypes, kernel_name):
    """Determines the queue where the kernel is to be launched.

    The execution queue is derived following Python Array API's
    "compute follows data" programming model.

    Args:
        argtypes : The Numba inferred type for each argument.
        kernel_name : The name of the kernel function

    Returns:
        A queue the common queue used to allocate the arrays. If no such
        queue exists, then raises an Exception.

    Raises:
        ExecutionQueueInferenceError: If the queue could not be inferred.
    """

    usmarray_argnums = [
        i for i, _ in enumerate(args) if isinstance(argtypes[i], USMNdArray)
    ]

    usm_array_args = [
        argtype for i, argtype in enumerate(argtypes) if i in usmarray_argnums
    ]

    queue = chk_compute_follows_data_compliance(usm_array_args)

    if not queue:
        raise ExecutionQueueInferenceError(
            kernel_name, usmarray_argnum_list=usmarray_argnums
        )

    return queue

1	# SPDX-FileCopyrightText: 2022 - 2024 Intel Corporation
2	#
3	# SPDX-License-Identifier: Apache-2.0
4
UNCOV 5	from numba_dpex.core.exceptions import ExecutionQueueInferenceError	×
UNCOV 6	from numba_dpex.core.types import USMNdArray	×
7
8
UNCOV 9	def chk_compute_follows_data_compliance(usm_array_arglist):	×
10	"""Check if all the usm ndarray's have the same device.
11
12	Extracts the device filter string from the Numba inferred USMNdArray
13	type. Check if the devices corresponding to the filter string are
14	equivalent and return a ``dpctl.SyclDevice`` object corresponding to the
15	common filter string.
16
17	If an exception occurred in creating a ``dpctl.SyclDevice``, or the
18	devices are not equivalent then returns None.
19
20	Args:
21	usm_array_arglist : A list of usm_ndarray types specified as
22	arguments to the kernel.
23
24	Returns:
25	A ``dpctl.SyclDevice`` object if all USMNdArray have same device, or
26	else None is returned.
27	"""
28
UNCOV 29	queue = None	×
30
UNCOV 31	for usm_array in usm_array_arglist:	×
UNCOV 32	_queue = usm_array.queue	×
UNCOV 33	if not queue:	×
UNCOV 34	queue = _queue	×
35	else:
UNCOV 36	if _queue != queue:	×
37	return None	×
38
UNCOV 39	return queue	×
40
41
UNCOV 42	def determine_kernel_launch_queue(args, argtypes, kernel_name):	×
43	"""Determines the queue where the kernel is to be launched.
44
45	The execution queue is derived following Python Array API's
46	"compute follows data" programming model.
47
48	Args:
49	argtypes : The Numba inferred type for each argument.
50	kernel_name : The name of the kernel function
51
52	Returns:
53	A queue the common queue used to allocate the arrays. If no such
54	queue exists, then raises an Exception.
55
56	Raises:
57	ExecutionQueueInferenceError: If the queue could not be inferred.
58	"""
59
UNCOV 60	usmarray_argnums = [	×
61	i for i, _ in enumerate(args) if isinstance(argtypes[i], USMNdArray)
62	]
63
UNCOV 64	usm_array_args = [	×
65	argtype for i, argtype in enumerate(argtypes) if i in usmarray_argnums
66	]
67
UNCOV 68	queue = chk_compute_follows_data_compliance(usm_array_args)	×
69
UNCOV 70	if not queue:	×
71	raise ExecutionQueueInferenceError(	×
72	kernel_name, usmarray_argnum_list=usmarray_argnums
73	)
74
UNCOV 75	return queue	×

IntelPython / numba-dpex / 8406344066

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