[Enhancement] Simplify kernel source extraction in JIT adapters

[LeiWang1999]

• Updated CtypesKernelAdapter and CythonKernelAdapter to include kernel_global_source for improved source code retrieval.
• Modified the _legalize_result_idx method in BaseKernelAdapter to accept Optional[List[int]] for better flexibility.
• Added comments to clarify the purpose of kernel_global_source in both adapters, enhancing code readability and maintainability.

[Copilot]

Pull Request Overview

This PR simplifies the extraction of kernel source code in the JIT adapters and makes the _legalize_result_idx interface more flexible.

• Updates both CtypesKernelAdapter and CythonKernelAdapter to pass a pre-extracted kernel_global_source.
• Modifies the constructor signatures and adjusts get_kernel_source methods accordingly.
• Updates BaseKernelAdapter’s _legalize_result_idx to accept an Optional[List[int]].

Reviewed Changes

Copilot reviewed 4 out of 4 changed files in this pull request and generated no comments.

File	Description
tilelang/jit/kernel.py	Extracts kernel_global_source for Ctypes and Cython backends for improved clarity.
tilelang/jit/adapter/ctypes/adapter.py	Refactors constructor to receive kernel_global_source; docstring still mentions rt_mod.
tilelang/jit/adapter/cython/adapter.py	Refactors constructor to receive kernel_global_source; docstring still mentions rt_mod.
tilelang/jit/adapter/base.py	Updates _legalize_result_idx signature for improved flexibility.

Comments suppressed due to low confidence (2)

tilelang/jit/adapter/ctypes/adapter.py:45

• The constructor's docstring still references 'rt_mod: Runtime module', which is no longer a parameter. Please update the docstring to reflect the new signature.

def __init__(self, params: List[TensorType], result_idx: List[int], target: str, func_or_mod: Union[tir.PrimFunc, tvm.IRModule], kernel_global_source: Optional[str] = None, verbose: bool = False, pass_configs: Optional[Dict[str, Any]] = None):

tilelang/jit/adapter/cython/adapter.py:155

• The docstring for the constructor still includes a reference to 'rt_mod: Runtime module'. Please update the docstring to remove the outdated parameter information.

def __init__(self, params: List[KernelParam], result_idx: List[int], target: Union[str, Target], func_or_mod: Union[tir.PrimFunc, tvm.IRModule], kernel_global_source: str, verbose: bool = False, pass_configs: Optional[Dict[str, Any]] = None):