Skip to main content

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__

__str__(self) -> String

Returns:

String

Was this page helpful?