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,
Copyable,
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]] = None, schedule: MatmulSchedule = MatmulSchedule(0), splits: OptionalReg[Int] = None, raster_order: OptionalReg[RasterOrder] = None) -> Self
__str__
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!