Improve python typing

Author: ZzEeKkAa

numba_dpex/core/kernel_interface/spirv_kernel.py

@@ -10,6 +10,7 @@
 from numba_dpex import config, spirv_generator
 from numba_dpex.core.compiler import compile_with_dpex
 from numba_dpex.core.exceptions import UncompiledKernelError, UnreachableError
+from numba_dpex.core.targets.kernel_target import DpexKernelTargetContext
 
 from .kernel_base import KernelInterface
 
@@ -133,9 +134,8 @@ def compile(
         )
 
         func = cres.library.get_function(cres.fndesc.llvm_func_name)
-        kernel = cres.target_context.prepare_spir_kernel(
-            func, cres.signature.args
-        )
+        kernel_targetctx: DpexKernelTargetContext = cres.target_context
+        kernel = kernel_targetctx.prepare_spir_kernel(func, cres.signature.args)
 
         # XXX: Setting the inline_threshold in the following way is a temporary
         # workaround till the JitKernel dispatcher is replaced by

numba_dpex/core/parfors/parfor_lowerer.py

@@ -185,7 +185,7 @@ def _submit_parfor_kernel(
         kl_builder.set_arguments(
             kernel_fn.kernel_arg_types, kernel_args=kernel_args
         )
-        kl_builder.set_dependant_event_list(dep_events=[])
+        kl_builder.set_dependant_event_list([])
         event_ref = kl_builder.submit()
 
         sycl.dpctl_event_wait(lowerer.builder, event_ref)

numba_dpex/core/targets/kernel_target.py

@@ -351,9 +351,8 @@ def mangler(self, name, argtypes, abi_tags=(), uid=None):
         )
 
     def prepare_spir_kernel(self, func, argtypes):
-        module = func.module
         func.linkage = "linkonce_odr"
-        module.data_layout = codegen.SPIR_DATA_LAYOUT[self.address_size]
+        func.module.data_layout = codegen.SPIR_DATA_LAYOUT[self.address_size]
         wrapper = self._generate_spir_kernel_wrapper(func, argtypes)
         return wrapper