tilelang.jit.adapter.utils module

tilelang.jit.adapter.utils.get_annotated_mod(func_or_mod: Union[PrimFunc, IRModule], target: Union[str, Target] = 'auto', target_host: Optional[Union[str, Target]] = None, model_type: Literal['device', 'host', 'all'] = 'all') → Union[IRModule, tuple[tvm.ir.module.IRModule, tvm.ir.module.IRModule]]

tilelang.jit.adapter.utils.is_cpu_target(target: Target) → bool

tilelang.jit.adapter.utils.is_cuda_target(target: Target) → bool

tilelang.jit.adapter.utils.is_hip_target(target: Target) → bool

tilelang.jit.adapter.utils.match_declare_kernel(source: str, annotation: str = '__global__') → int

tilelang.jit.adapter.utils.match_declare_kernel_cpu(source: str, annotation: str = 'int32_t') → int

tilelang.jit.adapter.utils.match_global_kernel(source: str, annotation: str = '__global__') → int