Remove default value for atomic ref index

IntelPython · Feb 27, 2024 · d3040d5 · d3040d5
1 parent a19cb7d
commit d3040d5
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/numba_dpex/experimental/_kernel_dpcpp_spirv_overloads/_atomic_ref_overloads.py b/numba_dpex/experimental/_kernel_dpcpp_spirv_overloads/_atomic_ref_overloads.py
@@ -584,7 +584,7 @@ def _check_if_supported_ref(ref):
 )
 def ol_atomic_ref(
     ref,
-    index=0,
+    index,
     memory_order=MemoryOrder.RELAXED,
     memory_scope=MemoryScope.DEVICE,
     address_space=AddressSpace.GLOBAL,
@@ -655,7 +655,7 @@ def ol_atomic_ref(
 
     def ol_atomic_ref_ctor_impl(
         ref,
-        index=0,
+        index,
         memory_order=MemoryOrder.RELAXED,  # pylint: disable=unused-argument
         memory_scope=MemoryScope.DEVICE,  # pylint: disable=unused-argument
         address_space=AddressSpace.GLOBAL,  # pylint: disable=unused-argument

diff --git a/numba_dpex/kernel_api/atomic_ref.py b/numba_dpex/kernel_api/atomic_ref.py
@@ -18,7 +18,7 @@ class AtomicRef:
     def __init__(  # pylint: disable=too-many-arguments
         self,
         ref,
-        index=0,
+        index,
         memory_order=MemoryOrder.RELAXED,
         memory_scope=MemoryScope.DEVICE,
         address_space=AddressSpace.GLOBAL,

diff --git a/numba_dpex/tests/experimental/kernel_api_overloads/spv_overloads/test_atomic_fence.py b/numba_dpex/tests/experimental/kernel_api_overloads/spv_overloads/test_atomic_fence.py
@@ -21,7 +21,7 @@ def test_atomic_fence():
     def _kernel(item: Item, a, b):
         i = item.get_id(0)
 
-        bref = AtomicRef(b)
+        bref = AtomicRef(b, index=0)
 
         if i == 1:
             a[i] += 1