7265827278

Committed 19 Dec 2023 06:09PM UTC coverage: 77.565% (-2.1%) from 79.663%

Build # 7265827278

Build Type

Pull #1252

github

Committed by

web-flow

Commit Message

Merge 9a0269560 into 6b76fd3de

Pull Request Pull Request #1252: WIP: Experimental/atomic ref

Run Details

1222 of 1848 branches covered (0.0%)

Branch coverage included in aggregate %.

125 of 322 new or added lines in 11 files covered. (38.82%)

2 existing lines in 2 files now uncovered.

5506 of 6826 relevant lines covered (80.66%)

0.81 hits per line

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

38.78

/numba_dpex/experimental/kernel_iface/atomic_ref.py

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

"""Implements a mock Python class to represent ``sycl::atomic_ref`` for
prototyping numba_dpex kernel functions before they are JIT compiled.
"""

from .memory_enums import AddressSpace, MemoryOrder, MemoryScope


class AtomicRef:
    """Analogue to the ``sycl::atomic_ref`` type. An atomic reference is a
    view into a data container that can be then updated atomically using any of
    the ``fetch_*`` member functions of the class.
    """

    def __init__(  # pylint: disable=too-many-arguments
        self,
        ref,
        index=0,
        memory_order=MemoryOrder.RELAXED,
        memory_scope=MemoryScope.DEVICE,
        address_space=AddressSpace.GLOBAL,
    ):
        """A Python stub to represent a SYCL AtomicRef class. An AtomicRef
        object represents a single element view into a Python array-like object
        that can be then updated using various fetch_* type functions.

        To maintain analogy with SYCL's API, the AtomicRef constructor takes
        optional ``memory_order``, ``memory_scope``, and ``address_space``
        arguments that are ignored in Python execution.
        """
        self._memory_order = memory_order
        self._memory_scope = memory_scope
        self._address_space = address_space
        self._ref = ref
        self._index = index

    @property
    def ref(self):
        """Returns the vlue stored in the AtomicRef._ref[_index]."""
        return self._ref[self._index]

    @property
    def memory_order(self) -> MemoryOrder:
        """Returns the MemoryOrder value used to define the AtomicRef."""
        return self._memory_order

    @property
    def memory_scope(self) -> MemoryScope:
        """Returns the MemoryScope value used to define the AtomicRef."""
        return self._memory_scope

    @property
    def address_space(self) -> AddressSpace:
        """Returns the AddressSpace value used to define the AtomicRef."""
        return self._address_space

    def fetch_add(self, val):
        """Adds the operand ``val`` to the object referenced by the AtomicRef
        and assigns the result to the value of the referenced object. Returns
        the original value of the object.

        Args:
            val : Value to be added to the object referenced by the AtomicRef.

        Returns: The original value of the object referenced by the AtomicRef.

        """
        old = self._ref[self._index]
        self._ref[self._index] += val
        return old

    def fetch_sub(self, val):
        """Subtracts the operand ``val`` to the object referenced by the
        AtomicRef and assigns the result to the value of the referenced object.
        Returns the original value of the object.

        Args:
            val : Value to be subtracted to the object referenced by the
            AtomicRef.

        Returns: The original value of the object referenced by the AtomicRef.

        """
        old = self._ref[self._index]
        self._ref[self._index] -= val
        return old

    def fetch_min(self, val):
        """Calculates the minimum value of the operand ``val`` and the object
        referenced by the AtomicRef and assigns the result to the value of the
        referenced object. Returns the original value of the object.

        Args:
            val : Value to be compared against to the object referenced by the
            AtomicRef.

        Returns: The original value of the object referenced by the AtomicRef.

        """
        old = self._ref[self._index]
        self._ref[self._index] = min(old, val)
        return old

    def fetch_max(self, val):
        """Calculates the maximum value of the operand ``val`` and the object
        referenced by the AtomicRef and assigns the result to the value of the
        referenced object. Returns the original value of the object.

        Args:
            val : Value to be compared against to the object referenced by the
            AtomicRef.

        Returns: The original value of the object referenced by the AtomicRef.

        """
        old = self._ref[self._index]
        self._ref[self._index] = max(old, val)
        return old

    def fetch_and(self, val):
        """Calculates the bitwise AND of the operand ``val`` and the object
        referenced by the AtomicRef and assigns the result to the value of the
        referenced object. Returns the original value of the object.

        Args:
            val : Value to be bitwise ANDed against to the object referenced by
            the AtomicRef.

        Returns: The original value of the object referenced by the AtomicRef.

        """
        old = self._ref[self._index]
        self._ref[self._index] &= val
        return old

    def fetch_or(self, val):
        """Calculates the bitwise OR of the operand ``val`` and the object
        referenced by the AtomicRef and assigns the result to the value of the
        referenced object. Returns the original value of the object.

        Args:
            val : Value to be bitwise ORed against to the object referenced by
            the AtomicRef.

        Returns: The original value of the object referenced by the AtomicRef.

        """
        old = self._ref[self._index]
        self._ref[self._index] |= val
        return old

    def fetch_xor(self, val):
        """Calculates the bitwise XOR of the operand ``val`` and the object
        referenced by the AtomicRef and assigns the result to the value of the
        referenced object. Returns the original value of the object.

        Args:
            val : Value to be bitwise XORed against to the object referenced by
            the AtomicRef.

        Returns: The original value of the object referenced by the AtomicRef.

        """
        old = self._ref[self._index]
        self._ref[self._index] ^= val
        return old

1	# SPDX-FileCopyrightText: 2023 Intel Corporation
2	#
3	# SPDX-License-Identifier: Apache-2.0
4
5	"""Implements a mock Python class to represent ``sycl::atomic_ref`` for	1✔
6	prototyping numba_dpex kernel functions before they are JIT compiled.
7	"""
8
9	from .memory_enums import AddressSpace, MemoryOrder, MemoryScope	1✔
10
11
12	class AtomicRef:	1✔
13	"""Analogue to the ``sycl::atomic_ref`` type. An atomic reference is a
14	view into a data container that can be then updated atomically using any of
15	the ``fetch_*`` member functions of the class.
16	"""
17
18	def __init__( # pylint: disable=too-many-arguments	1✔
19	self,
20	ref,
21	index=0,
22	memory_order=MemoryOrder.RELAXED,
23	memory_scope=MemoryScope.DEVICE,
24	address_space=AddressSpace.GLOBAL,
25	):
26	"""A Python stub to represent a SYCL AtomicRef class. An AtomicRef
27	object represents a single element view into a Python array-like object
28	that can be then updated using various fetch_* type functions.
29
30	To maintain analogy with SYCL's API, the AtomicRef constructor takes
31	optional ``memory_order``, ``memory_scope``, and ``address_space``
32	arguments that are ignored in Python execution.
33	"""
NEW 34	self._memory_order = memory_order	×
NEW 35	self._memory_scope = memory_scope	×
NEW 36	self._address_space = address_space	×
NEW 37	self._ref = ref	×
NEW 38	self._index = index	×
39
40	@property	1✔
41	def ref(self):	1✔
42	"""Returns the vlue stored in the AtomicRef._ref[_index]."""
NEW 43	return self._ref[self._index]	×
44
45	@property	1✔
46	def memory_order(self) -> MemoryOrder:	1✔
47	"""Returns the MemoryOrder value used to define the AtomicRef."""
NEW 48	return self._memory_order	×
49
50	@property	1✔
51	def memory_scope(self) -> MemoryScope:	1✔
52	"""Returns the MemoryScope value used to define the AtomicRef."""
NEW 53	return self._memory_scope	×
54
55	@property	1✔
56	def address_space(self) -> AddressSpace:	1✔
57	"""Returns the AddressSpace value used to define the AtomicRef."""
NEW 58	return self._address_space	×
59
60	def fetch_add(self, val):	1✔
61	"""Adds the operand ``val`` to the object referenced by the AtomicRef
62	and assigns the result to the value of the referenced object. Returns
63	the original value of the object.
64
65	Args:
66	val : Value to be added to the object referenced by the AtomicRef.
67
68	Returns: The original value of the object referenced by the AtomicRef.
69
70	"""
NEW 71	old = self._ref[self._index]	×
NEW 72	self._ref[self._index] += val	×
NEW 73	return old	×
74
75	def fetch_sub(self, val):	1✔
76	"""Subtracts the operand ``val`` to the object referenced by the
77	AtomicRef and assigns the result to the value of the referenced object.
78	Returns the original value of the object.
79
80	Args:
81	val : Value to be subtracted to the object referenced by the
82	AtomicRef.
83
84	Returns: The original value of the object referenced by the AtomicRef.
85
86	"""
NEW 87	old = self._ref[self._index]	×
NEW 88	self._ref[self._index] -= val	×
NEW 89	return old	×
90
91	def fetch_min(self, val):	1✔
92	"""Calculates the minimum value of the operand ``val`` and the object
93	referenced by the AtomicRef and assigns the result to the value of the
94	referenced object. Returns the original value of the object.
95
96	Args:
97	val : Value to be compared against to the object referenced by the
98	AtomicRef.
99
100	Returns: The original value of the object referenced by the AtomicRef.
101
102	"""
NEW 103	old = self._ref[self._index]	×
NEW 104	self._ref[self._index] = min(old, val)	×
NEW 105	return old	×
106
107	def fetch_max(self, val):	1✔
108	"""Calculates the maximum value of the operand ``val`` and the object
109	referenced by the AtomicRef and assigns the result to the value of the
110	referenced object. Returns the original value of the object.
111
112	Args:
113	val : Value to be compared against to the object referenced by the
114	AtomicRef.
115
116	Returns: The original value of the object referenced by the AtomicRef.
117
118	"""
NEW 119	old = self._ref[self._index]	×
NEW 120	self._ref[self._index] = max(old, val)	×
NEW 121	return old	×
122
123	def fetch_and(self, val):	1✔
124	"""Calculates the bitwise AND of the operand ``val`` and the object
125	referenced by the AtomicRef and assigns the result to the value of the
126	referenced object. Returns the original value of the object.
127
128	Args:
129	val : Value to be bitwise ANDed against to the object referenced by
130	the AtomicRef.
131
132	Returns: The original value of the object referenced by the AtomicRef.
133
134	"""
NEW 135	old = self._ref[self._index]	×
NEW 136	self._ref[self._index] &= val	×
NEW 137	return old	×
138
139	def fetch_or(self, val):	1✔
140	"""Calculates the bitwise OR of the operand ``val`` and the object
141	referenced by the AtomicRef and assigns the result to the value of the
142	referenced object. Returns the original value of the object.
143
144	Args:
145	val : Value to be bitwise ORed against to the object referenced by
146	the AtomicRef.
147
148	Returns: The original value of the object referenced by the AtomicRef.
149
150	"""
NEW 151	old = self._ref[self._index]	×
NEW 152	self._ref[self._index] \|= val	×
NEW 153	return old	×
154
155	def fetch_xor(self, val):	1✔
156	"""Calculates the bitwise XOR of the operand ``val`` and the object
157	referenced by the AtomicRef and assigns the result to the value of the
158	referenced object. Returns the original value of the object.
159
160	Args:
161	val : Value to be bitwise XORed against to the object referenced by
162	the AtomicRef.
163
164	Returns: The original value of the object referenced by the AtomicRef.
165
166	"""
NEW 167	old = self._ref[self._index]	×
NEW 168	self._ref[self._index] ^= val	×
NEW 169	return old	×

IntelPython / numba-dpex / 7265827278

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