Changes to core.runtime.context.py

   - Improve few docstrins.
   - Change the type signature of DPEXRT_MemInfo_fill to use
     an llvmlite.ir.InteValue(64) instead of cguitls.intp_t

Author: Diptorup Deb

numba_dpex/core/runtime/context.py

@@ -30,9 +30,8 @@ def wrap(self, builder, *args, **kwargs):
     @_check_null_result
     def meminfo_alloc(self, builder, size, usm_type, device):
         """
-        A wrapped caller for :func:`~context.DpexRTContext.meminfo_alloc_unchecked`
-        with null check. Please refer to that function for the details on how the
-        null check is done.
+        Wrapper to call :func:`~context.DpexRTContext.meminfo_alloc_unchecked`
+        with null checking of the returned value.
         """
         return self.meminfo_alloc_unchecked(builder, size, usm_type, device)
 
@@ -48,9 +47,8 @@ def meminfo_fill(
         device,
     ):
         """
-        A wrapped caller for :func:`~context.DpexRTContext.meminfo_fill_unchecked`
-        with null check. Please refer to that function for the details on how the
-        null check is done.
+        Wrapper to call :func:`~context.DpexRTContext.meminfo_fill_unchecked`
+        with null checking of the returned value.
         """
         return self.meminfo_fill_unchecked(
             builder,
@@ -126,7 +124,7 @@ def meminfo_fill_unchecked(
         b = llvmir.IntType(1)
         fnty = llvmir.FunctionType(
             cgutils.voidptr_t,
-            [cgutils.voidptr_t, u64, b, b, cgutils.intp_t, cgutils.voidptr_t],
+            [cgutils.voidptr_t, u64, b, b, u64, cgutils.voidptr_t],
         )
         fn = cgutils.get_or_insert_function(mod, fnty, "DPEXRT_MemInfo_fill")
         fn.return_value.add_attribute("noalias")