Wrap const indexing overloads into function

Author: ZzEeKkAa

numba_dpex/experimental/_kernel_dpcpp_spirv_overloads/_index_space_id_overloads.py

@@ -180,26 +180,31 @@ def ol_item_get_index_impl(item, dim):
     return ol_item_gen_index
 
 
-_index_const_overload_methods = [
-    (ItemType, "get_id", _intrinsic_spirv_global_invocation_id),
-    (ItemType, "get_range", _intrinsic_spirv_global_size),
-    (NdItemType, "get_global_id", _intrinsic_spirv_global_invocation_id),
-    (NdItemType, "get_local_id", _intrinsic_spirv_local_invocation_id),
-    (NdItemType, "get_global_range", _intrinsic_spirv_global_size),
-    (NdItemType, "get_local_range", _intrinsic_spirv_workgroup_size),
-    (GroupType, "get_group_id", _intrinsic_spirv_workgroup_id),
-    (GroupType, "get_group_range", _intrinsic_spirv_numworkgroups),
-    (GroupType, "get_local_range", _intrinsic_spirv_workgroup_size),
-]
-
-for index_overload in _index_const_overload_methods:
-    _type, method, _intrinsic = index_overload
-
-    ol_index_func = generate_index_overload(_type, _intrinsic)
-
-    overload_method(_type, method, target=DPEX_KERNEL_EXP_TARGET_NAME)(
-        ol_index_func
-    )
+def register_index_const_methods():
+    """Register indexing related methods that can be defined as spirv const."""
+    _index_const_overload_methods = [
+        (ItemType, "get_id", _intrinsic_spirv_global_invocation_id),
+        (ItemType, "get_range", _intrinsic_spirv_global_size),
+        (NdItemType, "get_global_id", _intrinsic_spirv_global_invocation_id),
+        (NdItemType, "get_local_id", _intrinsic_spirv_local_invocation_id),
+        (NdItemType, "get_global_range", _intrinsic_spirv_global_size),
+        (NdItemType, "get_local_range", _intrinsic_spirv_workgroup_size),
+        (GroupType, "get_group_id", _intrinsic_spirv_workgroup_id),
+        (GroupType, "get_group_range", _intrinsic_spirv_numworkgroups),
+        (GroupType, "get_local_range", _intrinsic_spirv_workgroup_size),
+    ]
+
+    for index_overload in _index_const_overload_methods:
+        _type, method, _intrinsic = index_overload
+
+        ol_index_func = generate_index_overload(_type, _intrinsic)
+
+        overload_method(_type, method, target=DPEX_KERNEL_EXP_TARGET_NAME)(
+            ol_index_func
+        )
+
+
+register_index_const_methods()
 
 
 @intrinsic(target=DPEX_KERNEL_EXP_TARGET_NAME)