tilelang.jit.adapter.utils ========================== .. py:module:: tilelang.jit.adapter.utils Functions --------- .. autoapisummary:: tilelang.jit.adapter.utils.match_global_kernel tilelang.jit.adapter.utils.match_declare_kernel tilelang.jit.adapter.utils.match_declare_kernel_cpu tilelang.jit.adapter.utils.is_cuda_target tilelang.jit.adapter.utils.is_hip_target tilelang.jit.adapter.utils.is_cpu_target tilelang.jit.adapter.utils.get_annotated_mod tilelang.jit.adapter.utils.pythonic_expr Module Contents --------------- .. py:function:: match_global_kernel(source, annotation = '__global__') .. py:function:: match_declare_kernel(source, annotation = '__global__') .. py:function:: match_declare_kernel_cpu(source, annotation = 'int32_t') .. py:function:: is_cuda_target(target) .. py:function:: is_hip_target(target) .. py:function:: is_cpu_target(target) .. py:function:: get_annotated_mod(func_or_mod, target = 'auto', target_host = None, model_type = 'all') .. py:function:: pythonic_expr(expr, dtype_map = None) Converts a TVM PrimExpr into a Python-style string, correctly handling operator precedence. :param expr: The TVM PrimExpr to convert. :returns: A string representation of the expression.