Mojo struct
TuningConfigSM90
struct TuningConfigSM90
Fields
- M (
Int): - N (
Int): - K (
Int): - mma_shape (
IndexList[3]): - block_tile_shape (
IndexList[3]): - num_pipeline_stages (
Int): - cluster_shape (
IndexList[3]): - num_consumer (
Int): - partitioned_multicast (
Bool): - grid_shape (
OptionalReg[IndexList[2]]): - schedule (
MatmulSchedule): - splits (
OptionalReg[Int]): - raster_order (
OptionalReg[RasterOrder]):
Implemented traits
AnyType,
Copyable,
ImplicitlyCopyable,
ImplicitlyDestructible,
Movable,
RegisterPassable,
TrivialRegisterPassable,
TuningConfig,
Writable
Methods
__init__
__init__(M: Int, N: Int, K: Int, mma_shape: IndexList[3], block_tile_shape: IndexList[3], num_pipeline_stages: Int, cluster_shape: IndexList[3], num_consumer: Int, partitioned_multicast: Bool, grid_shape: OptionalReg[IndexList[2]] = None, schedule: MatmulSchedule = MatmulSchedule.NONE, splits: OptionalReg[Int] = None, raster_order: OptionalReg[RasterOrder] = None) -> Self
write_to
write_to(self, mut writer: T)
Writes the tuning config as a string.
Args:
- writer (
T): The writer to write to.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!