tilelang.carver package

Subpackages

• tilelang.carver.arch package
  • Subpackages
    • tilelang.carver.arch.driver package
      • Submodules
      • Module contents
  • Submodules
    • tilelang.carver.arch.arch_base module
      • TileDevice
    • tilelang.carver.arch.cdna module
      • CDNA
      • is_cdna_arch()
    • tilelang.carver.arch.cpu module
      • CPU
      • is_cpu_arch()
    • tilelang.carver.arch.cuda module
      • CUDA
      • TensorInstruction
      • check_sm_version()
      • has_mma_support()
      • is_ada_arch()
      • is_ampere_arch()
      • is_cuda_arch()
      • is_hopper_arch()
      • is_tensorcore_supported_precision()
      • is_volta_arch()
  • Module contents
    • auto_infer_current_arch()
    • get_arch()
• tilelang.carver.roller package
  • Subpackages
    • tilelang.carver.roller.policy package
      • Submodules
      • Module contents
    • tilelang.carver.roller.shape_inference package
      • Submodules
      • Module contents
  • Submodules
    • tilelang.carver.roller.bestfit module
      • BestFit
      • Block
    • tilelang.carver.roller.hint module
      • Hint
      • IntrinInfo
      • Stride
      • TensorCoreExtraConfig
      • TileDict
    • tilelang.carver.roller.node module
      • BlockAnalyzer
      • Edge
      • Node
      • OutputNode
      • PlaceHolderNode
      • PrimFuncNode
      • find_topo_sort()
      • find_topo_sort_priority()
      • pre_order_traverse()
      • topo_order()
    • tilelang.carver.roller.rasterization module
      • NoRasterization
      • Rasterization
      • Rasterization2DColumn
      • Rasterization2DRow
  • Module contents
• tilelang.carver.template package
  • Submodules
    • tilelang.carver.template.base module
      • BaseTemplate
    • tilelang.carver.template.conv module
      • ConvTemplate
    • tilelang.carver.template.elementwise module
      • ElementwiseTemplate
    • tilelang.carver.template.flashattention module
      • FlashAttentionTemplate
    • tilelang.carver.template.gemv module
      • GEMVTemplate
    • tilelang.carver.template.general_reduce module
      • GeneralReductionTemplate
    • tilelang.carver.template.matmul module
      • MatmulTemplate
  • Module contents

Submodules

• tilelang.carver.analysis module
  • BlockInfo
    • BlockInfo.block_rv
    • BlockInfo.dom()
    • BlockInfo.dom_kind()
    • BlockInfo.is_elementwise()
    • BlockInfo.is_gemm()
    • BlockInfo.is_gemv()
    • BlockInfo.is_injective()
    • BlockInfo.is_reduction()
    • BlockInfo.iters
    • BlockInfo.name
  • IterInfo
    • IterInfo.dom
    • IterInfo.kind
    • IterInfo.loop_rv
    • IterInfo.var
  • check_func_with_dynamic()
  • collect_block_iter_vars_used_in_access_region()
  • collect_vars_used_in_prim_expr()
  • detect_dominant_read()
  • find_var_from_func()
  • get_coalesced_veclen()
  • get_max_shared_memory_per_block()
  • get_max_threads_per_block()
  • get_reduction_blocks()
  • get_root_block()
  • is_broadcast_epilogue()
  • normalize_prim_func()
• tilelang.carver.common_schedules module
  • get_block()
  • get_output_blocks()
  • try_inline()
  • try_inline_contiguous_spatial()
• tilelang.carver.matmul_analysis module
  • IterKind
    • IterKind.kIter_I
    • IterKind.kIter_J
    • IterKind.kIter_K
    • IterKind.kIter_S
    • IterKind.kIter_T
  • IterTrait
    • IterTrait.extent
    • IterTrait.kind
  • auto_inline_consumer_chain()
  • auto_inline_consumers()
  • auto_inline_producers()
  • collect_vars_from_expr()
  • detect_iter_traits()
  • find_arg_idx_from_buffer_chain()
  • find_first_similar_buffer()
  • find_first_similar_region()
  • find_last_producer_from_buffer()
  • get_dequantize_block()
  • get_in_out_dtypes()
  • get_index_map()
  • get_ladder_stage3_map()
  • get_propagate_map()
  • get_tensorized_func_and_tags()
  • inline_transpose_block()
  • is_identity_block()
  • is_identity_or_transpose_block()
  • is_transpose_block()
  • layout_propagate_chain()
  • make_iter_fusion_index_map()
  • normalize_to_matmul()
• tilelang.carver.utils module
  • get_rasterization_code()
  • get_roller_hints_from_func()
  • get_roller_hints_from_output_nodes()
  • retrieve_func_from_module()

Module contents

Base infra