tilelang.carver.roller package

Subpackages

• tilelang.carver.roller.policy package
  • Submodules
    • tilelang.carver.roller.policy.common module
      • coalesced_factor()
      • coalesced_tensor_shape()
      • factorize()
      • get_all_factors()
    • tilelang.carver.roller.policy.default module
      • DefaultPolicy
    • tilelang.carver.roller.policy.tensorcore module
      • TensorCorePolicy
  • Module contents
• tilelang.carver.roller.shape_inference package
  • Submodules
    • tilelang.carver.roller.shape_inference.common module
      • InputShapeInference
      • Statement
    • tilelang.carver.roller.shape_inference.tir module
      • DependencyAnalysis
      • InputShapeInference
      • Statement
      • TensorDepNode
      • get_analyzer_by_tir()
      • region_exist_in_list()
      • walk_indice()
  • Module contents

Submodules

• tilelang.carver.roller.bestfit module
  • BestFit
    • BestFit.free()
    • BestFit.malloc()
  • Block
    • Block.merge()
    • Block.size()
• tilelang.carver.roller.hint module
  • Hint
    • Hint.complete_config()
    • Hint.from_dict()
    • Hint.raxis_order
    • Hint.step
    • Hint.tensorcore_legalization()
    • Hint.to_dict()
  • IntrinInfo
    • IntrinInfo.inter_transform_a
    • IntrinInfo.inter_transform_b
    • IntrinInfo.is_input_8bit()
    • IntrinInfo.smooth_a
    • IntrinInfo.smooth_b
  • Stride
    • Stride.ax
    • Stride.compute_elements_from_shape()
    • Stride.compute_strides_from_shape()
    • Stride.is_valid()
    • Stride.stride
  • TensorCoreExtraConfig
  • TileDict
    • TileDict.get_rstep()
    • TileDict.get_tile()
• tilelang.carver.roller.node module
  • BlockAnalyzer
    • BlockAnalyzer.get_block_info()
    • BlockAnalyzer.get_block_name()
    • BlockAnalyzer.get_buffers()
    • BlockAnalyzer.get_consumer_blocks()
    • BlockAnalyzer.get_input_buffers()
    • BlockAnalyzer.get_output_buffers()
    • BlockAnalyzer.get_producer_blocks()
    • BlockAnalyzer.get_reduce_axis()
    • BlockAnalyzer.get_spatial_axis()
  • Edge
    • Edge.dst_id
    • Edge.dst_node
    • Edge.src_id
    • Edge.src_node
  • Node
    • Node.add_tag()
    • Node.get_dtype()
    • Node.get_ir()
    • Node.get_shape()
    • Node.get_tag()
    • Node.inputs
    • Node.is_output()
    • Node.is_placeholder()
    • Node.num_outputs()
    • Node.outputs
    • Node.set_dtype()
    • Node.set_inputs()
    • Node.set_outputs()
    • Node.set_shape()
    • Node.set_tag()
    • Node.update_tags()
  • OutputNode
    • OutputNode.get_ir()
    • OutputNode.is_output()
  • PlaceHolderNode
    • PlaceHolderNode.get_ir()
    • PlaceHolderNode.is_placeholder()
  • PrimFuncNode
    • PrimFuncNode.extent_wrapper()
    • PrimFuncNode.footprint()
    • PrimFuncNode.get_buffer_dtype()
    • PrimFuncNode.get_input_buffers()
    • PrimFuncNode.get_opt_shape()
    • PrimFuncNode.get_reduce_inputs_dtype()
    • PrimFuncNode.get_space_dim()
    • PrimFuncNode.infer_tensorcore_axis()
    • PrimFuncNode.propagate()
    • PrimFuncNode.propagate_inputs()
    • PrimFuncNode.propagate_inputs_on_reduction()
    • PrimFuncNode.propagate_outputs()
    • PrimFuncNode.propagate_reduction_inputs()
    • PrimFuncNode.set_dtype()
  • find_topo_sort()
  • find_topo_sort_priority()
  • pre_order_traverse()
  • topo_order()
• tilelang.carver.roller.rasterization module
  • NoRasterization
    • NoRasterization.get_code()
  • Rasterization
    • Rasterization.get_code()
    • Rasterization.panel_width
    • Rasterization.panel_width_
  • Rasterization2DColumn
    • Rasterization2DColumn.get_code()
    • Rasterization2DColumn.get_device_function()
  • Rasterization2DRow
    • Rasterization2DRow.get_code()

Module contents