5050215003

Build Type

push

github

Committed by GitHub

Commit Message

Merge pull request #1049 from IntelPython/feature/remove_numpy_array_kernelargs

Run Details

1042 of 1604 branches covered (64.96%)

Branch coverage included in aggregate %.

30 of 33 new or added lines in 6 files covered. (90.91%)

5 existing lines in 2 files now uncovered.

4397 of 5546 relevant lines covered (79.28%)

0.79 hits per line

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

86.67

/numba_dpex/core/kernel_interface/utils.py

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

IntelPython / numba-dpex / 5050215003

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