4768117657

Build Type

github

Committed by GitHub

Commit Message

Merge c81f81689 into 397178824

Pull Request Pull Request #946: Using cached queue instead of creating new one on type inference

Run Details

1908 of 3553 branches covered (53.7%)

Branch coverage included in aggregate %.

11 of 14 new or added lines in 2 files covered. (78.57%)

1 existing line in 1 file now uncovered.

6705 of 10175 relevant lines covered (65.9%)

0.66 hits per line

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

96.1

/numba_dpex/core/utils/suai_helper.py

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

import logging

import dpctl
import dpctl.memory as dpctl_mem
import numpy as np


class SyclUSMArrayInterface:
    """Stores as attributes the information extracted from a
    __sycl_usm_array_interface__ dictionary as defined by dpctl.memory.Memory*
    classes.
    """

    def __init__(
        self,
        data,
        writable,
        size,
        shape,
        dimensions,
        itemsize,
        strides,
        dtype,
        usm_type,
        device,
        queue,
    ):
        self._data = data
        self._data_writeable = writable
        self._size = size
        self._shape = shape
        self._dimensions = dimensions
        self._itemsize = itemsize
        self._strides = strides
        self._dtype = dtype
        self._usm_type = usm_type
        self._device = device
        self._queue = queue

    @property
    def data(self):
        return self._data

    @property
    def is_writable(self):
        return self._data_writeable

    @property
    def size(self):
        return self._size

    @property
    def shape(self):
        return self._shape

    @property
    def dimensions(self):
        return self._dimensions

    @property
    def itemsize(self):
        return self._itemsize

    @property
    def strides(self):
        return self._strides

    @property
    def dtype(self):
        return self._dtype

    @property
    def usm_type(self):
        return self._usm_type

    @property
    def device(self):
        return self._device

    @property
    def queue(self):
        return self._queue


def get_info_from_suai(obj):
    """
    Extracts the metadata of an object of type UsmNdArray using the objects
    __sycl_usm_array_interface__ (SUAI) attribute.

    The ``dpctl.memory.as_usm_memory`` function converts the array-like
    object into a dpctl.memory.USMMemory object. Using the ``as_usm_memory``
    is an implicit way to verify if the array-like object is a legal
    SYCL USM memory back Python object that can be passed to a dpex kernel.

    Args:
        obj: array-like object with a SUAI attribute.

    Returns:
        A SyclUSMArrayInterface object

    """

    # dpctl.as_usm_memory validated if an array-like object, obj, has a well
    # defined __sycl_usm_array_interface__ dictionary and converts it into a
    # dpctl.memory.Memory* object.
    try:
        usm_mem = dpctl_mem.as_usm_memory(obj)
    except Exception as e:
        logging.exception(
            "Array like object with __sycl_usm_array_interface__ could not be "
            "converted to a dpctl.memory.Memory* object."
        )
        raise e

    # The data attribute of __sycl_usm_array_interface__ is a 2-tuple.
    # The first element is the data pointer and the second a boolean
    # value indicating if the data is writable.
    is_writable = usm_mem.__sycl_usm_array_interface__["data"][1]

    shape = obj.__sycl_usm_array_interface__["shape"]
    total_size = np.prod(shape)
    ndim = len(shape)
    dtype = np.dtype(obj.__sycl_usm_array_interface__["typestr"])
    itemsize = dtype.itemsize

    strides = obj.__sycl_usm_array_interface__["strides"]
    if strides is None:
        strides = [1] * ndim
        for i in reversed(range(1, ndim)):
            strides[i - 1] = strides[i] * shape[i]
        strides = tuple(strides)

    syclobj = usm_mem.sycl_queue
    device = syclobj.sycl_device.filter_string
    usm_type = usm_mem.get_usm_type()

    suai_info = SyclUSMArrayInterface(
        data=usm_mem,
        writable=is_writable,
        size=total_size,
        usm_type=usm_type,
        device=device,
        queue=syclobj,
        shape=shape,
        dimensions=ndim,
        itemsize=itemsize,
        strides=strides,
        dtype=dtype,
    )

    return suai_info

1	# SPDX-FileCopyrightText: 2020 - 2023 Intel Corporation
2	#
3	# SPDX-License-Identifier: Apache-2.0
4
5	import logging	1✔
6
7	import dpctl	1✔
8	import dpctl.memory as dpctl_mem	1✔
9	import numpy as np	1✔
10
11
12	class SyclUSMArrayInterface:	1✔
13	"""Stores as attributes the information extracted from a
14	__sycl_usm_array_interface__ dictionary as defined by dpctl.memory.Memory*
15	classes.
16	"""
17
18	def __init__(	1✔
19	self,
20	data,
21	writable,
22	size,
23	shape,
24	dimensions,
25	itemsize,
26	strides,
27	dtype,
28	usm_type,
29	device,
30	queue,
31	):
32	self._data = data	1✔
33	self._data_writeable = writable	1✔
34	self._size = size	1✔
35	self._shape = shape	1✔
36	self._dimensions = dimensions	1✔
37	self._itemsize = itemsize	1✔
38	self._strides = strides	1✔
39	self._dtype = dtype	1✔
40	self._usm_type = usm_type	1✔
41	self._device = device	1✔
42	self._queue = queue	1✔
43
44	@property	1✔
45	def data(self):	1✔
46	return self._data	1✔
47
48	@property	1✔
49	def is_writable(self):	1✔
50	return self._data_writeable	1✔
51
52	@property	1✔
53	def size(self):	1✔
54	return self._size	1✔
55
56	@property	1✔
57	def shape(self):	1✔
58	return self._shape	1✔
59
60	@property	1✔
61	def dimensions(self):	1✔
62	return self._dimensions	1✔
63
64	@property	1✔
65	def itemsize(self):	1✔
66	return self._itemsize	1✔
67
68	@property	1✔
69	def strides(self):	1✔
70	return self._strides	1✔
71
72	@property	1✔
73	def dtype(self):	1✔
74	return self._dtype	1✔
75
76	@property	1✔
77	def usm_type(self):	1✔
78	return self._usm_type	1✔
79
80	@property	1✔
81	def device(self):	1✔
UNCOV 82	return self._device	×
83
84	@property	1✔
85	def queue(self):	1✔
86	return self._queue	1✔
87
88
89	def get_info_from_suai(obj):	1✔
90	"""
91	Extracts the metadata of an object of type UsmNdArray using the objects
92	__sycl_usm_array_interface__ (SUAI) attribute.
93
94	The ``dpctl.memory.as_usm_memory`` function converts the array-like
95	object into a dpctl.memory.USMMemory object. Using the ``as_usm_memory``
96	is an implicit way to verify if the array-like object is a legal
97	SYCL USM memory back Python object that can be passed to a dpex kernel.
98
99	Args:
100	obj: array-like object with a SUAI attribute.
101
102	Returns:
103	A SyclUSMArrayInterface object
104
105	"""
106
107	# dpctl.as_usm_memory validated if an array-like object, obj, has a well
108	# defined __sycl_usm_array_interface__ dictionary and converts it into a
109	# dpctl.memory.Memory* object.
110	try:	1✔
111	usm_mem = dpctl_mem.as_usm_memory(obj)	1✔
112	except Exception as e:	1✔
113	logging.exception(	1✔
114	"Array like object with __sycl_usm_array_interface__ could not be "
115	"converted to a dpctl.memory.Memory* object."
116	)
117	raise e	×
118
119	# The data attribute of __sycl_usm_array_interface__ is a 2-tuple.
120	# The first element is the data pointer and the second a boolean
121	# value indicating if the data is writable.
122	is_writable = usm_mem.__sycl_usm_array_interface__["data"][1]	1✔
123
124	shape = obj.__sycl_usm_array_interface__["shape"]	1✔
125	total_size = np.prod(shape)	1✔
126	ndim = len(shape)	1✔
127	dtype = np.dtype(obj.__sycl_usm_array_interface__["typestr"])	1✔
128	itemsize = dtype.itemsize	1✔
129
130	strides = obj.__sycl_usm_array_interface__["strides"]	1✔
131	if strides is None:	1!
132	strides = [1] * ndim	1✔
133	for i in reversed(range(1, ndim)):	1✔
134	strides[i - 1] = strides[i] * shape[i]	1✔
135	strides = tuple(strides)	1✔
136
137	syclobj = usm_mem.sycl_queue	1✔
138	device = syclobj.sycl_device.filter_string	1✔
139	usm_type = usm_mem.get_usm_type()	1✔
140
141	suai_info = SyclUSMArrayInterface(	1✔
142	data=usm_mem,
143	writable=is_writable,
144	size=total_size,
145	usm_type=usm_type,
146	device=device,
147	queue=syclobj,
148	shape=shape,
149	dimensions=ndim,
150	itemsize=itemsize,
151	strides=strides,
152	dtype=dtype,
153	)
154
155	return suai_info	1✔

IntelPython / numba-dpex / 4768117657

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