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
    • DefaultPolicy.arch
    • DefaultPolicy.assign_block_size()
    • DefaultPolicy.check_tile_shape_isvalid()
    • DefaultPolicy.compute_node_stride_map()
    • DefaultPolicy.compute_tile_dict()
    • DefaultPolicy.compute_workload_per_item()
    • DefaultPolicy.dfs_smem_tile()
    • DefaultPolicy.emit_config()
    • DefaultPolicy.from_output_nodes()
    • DefaultPolicy.from_prim_func()
    • DefaultPolicy.func
    • DefaultPolicy.get_base_tile()
    • DefaultPolicy.get_block_size()
    • DefaultPolicy.get_node_reduce_step_candidates()
    • DefaultPolicy.infer_node_smem_usage()
    • DefaultPolicy.nodes
    • DefaultPolicy.plan_rasterization()
    • DefaultPolicy.recommend_block_size()
    • DefaultPolicy.score_block_size()
    • DefaultPolicy.tags
• tilelang.carver.roller.policy.tensorcore module
  • TensorCorePolicy
    • TensorCorePolicy.block_reduction_depth
    • TensorCorePolicy.check_tile_shape_isvalid()
    • TensorCorePolicy.compute_node_stride_map()
    • TensorCorePolicy.get_node_reduce_step_candidates()
    • TensorCorePolicy.infer_node_smem_usage()
    • TensorCorePolicy.pipeline_stage
    • TensorCorePolicy.plan_rasterization()
    • TensorCorePolicy.use_async_copy
    • TensorCorePolicy.wmma_k

Module contents