Mojo struct
TuningConfigSM90
@register_passable(trivial)
struct TuningConfigSM90
Fields
- M (
Int
): - N (
Int
): - K (
Int
): - mma_shape (
IndexList[3]
): - block_tile_shape (
IndexList[3]
): - num_pipeline_stages (
UInt
): - cluster_shape (
IndexList[3]
): - num_consumer (
UInt
): - partitioned_multicast (
Bool
): - grid_shape (
OptionalReg[IndexList[2]]
): - schedule (
MatmulSchedule
): - splits (
OptionalReg[Int]
): - raster_order (
OptionalReg[RasterOrder]
):
Implemented traits
AnyType
,
ExplicitlyCopyable
,
ImplicitlyCopyable
,
Movable
,
Stringable
,
TuningConfig
,
UnknownDestructibility
Aliases
__copyinit__is_trivial
alias __copyinit__is_trivial = True
__del__is_trivial
alias __del__is_trivial = True
__moveinit__is_trivial
alias __moveinit__is_trivial = True
Methods
__init__
__init__(M: Int, N: Int, K: Int, mma_shape: IndexList[3], block_tile_shape: IndexList[3], num_pipeline_stages: UInt, cluster_shape: IndexList[3], num_consumer: UInt, partitioned_multicast: Bool, grid_shape: OptionalReg[IndexList[2]] = OptionalReg[IndexList[2]]({:i1 0, 1}), schedule: MatmulSchedule = MatmulSchedule(-1), splits: OptionalReg[Int] = OptionalReg[Int]({:i1 0, 1}), raster_order: OptionalReg[RasterOrder] = OptionalReg[RasterOrder]({:i1 0, 1})) -> Self
__str__
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!