For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /max/get-started.md).
Mojo struct
TileScheduler
struct TileScheduler[num_stages: Int, cluster_shape: IndexList[Int(3), element_type=DType.uint32] = Index[Int, Int, Int, dtype=DType.uint32](Int(1), Int(1), Int(1)), rasterize_order: RasterOrder = RasterOrder.AlongM, block_swizzle_size: Int = Int(8)]
Fields
- cluster_dim (
StaticTuple[Int32, Int(3)]): - log_cluster_dim_m (
FastDiv[DType.uint32]): - log_cluster_dim_n (
FastDiv[DType.uint32]): - log_cluster_dim_k (
FastDiv[DType.uint32]): - clc_response (
UnsafePointer[UInt128, MutAnyOrigin, address_space=AddressSpace.SHARED]): - full_mbar (
UnsafePointer[SharedMemBarrier, MutAnyOrigin, address_space=AddressSpace.SHARED]): - empty_mbar (
UnsafePointer[SharedMemBarrier, MutAnyOrigin, address_space=AddressSpace.SHARED]):
Implemented traits
AnyType,
Copyable,
ImplicitlyCopyable,
ImplicitlyDeletable,
Movable,
RegisterPassable,
TrivialRegisterPassable
comptime members
cluster_size
comptime cluster_size = (Int((mul cluster_shape[Int(0)], cluster_shape[Int(1)])) * cluster_shape[Int(2)])
log_cluster_k
comptime log_cluster_k = FastDiv(cluster_shape[Int(2)])
log_cluster_m
comptime log_cluster_m = FastDiv(cluster_shape[Int(0)])
log_cluster_n
comptime log_cluster_n = FastDiv(cluster_shape[Int(1)])
Methods
__init__
def __init__(cluster_dim: StaticTuple[Int32, Int(3)], clc_response_ptr: UnsafePointer[UInt128, address_space=AddressSpace.SHARED], full_mbar_ptr: UnsafePointer[SharedMemBarrier, address_space=AddressSpace.SHARED], empty_mbar_ptr: UnsafePointer[SharedMemBarrier, address_space=AddressSpace.SHARED]) -> Self
work_info_from_clc_response
static def work_info_from_clc_response(result: UnsafePointer[UInt128, address_space=AddressSpace.SHARED]) -> WorkInfo
Returns:
work_info_from_cluster
static def work_info_from_cluster(work_info: WorkInfo, cluster_dim: StaticTuple[Int32, Int(3)], log_cluster_dim_m: FastDiv[DType.uint32], log_cluster_dim_n: FastDiv[DType.uint32]) -> WorkInfo
Returns:
initial_work_info
fetch_next_work
def fetch_next_work(self, work_info: WorkInfo, consumer_state: PipelineState[num_stages]) -> WorkInfo
Returns:
advance_to_next_work
def advance_to_next_work(self, mut clc_state: PipelineState[num_stages]) -> PipelineState[num_stages]
Returns:
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!