Mojo struct
TuningConfigSM100
struct TuningConfigSM100
Fields
- M (
Int): - M_end (
Int): - N (
Int): - K (
Int): - mma_shape (
IndexList[3]): - block_tile_shape (
IndexList[3]): - cluster_shape (
IndexList[3]): - block_swizzle_size (
Int): - rasterize_order (
RasterOrder): - cta_group (
Int): - swapAB (
Bool): - k_group_size (
Int): - num_accum_pipeline_stages (
Int): - num_clc_pipeline_stages (
Int): - num_split_k (
Int): - is_small_bn (
Bool):
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], cluster_shape: IndexList[3], block_swizzle_size: Int, rasterize_order: RasterOrder, cta_group: Int = 2, swapAB: Bool = False, k_group_size: Int = 1, num_accum_pipeline_stages: Int = 2, num_clc_pipeline_stages: Int = 2, num_split_k: Int = 1, is_small_bn: Bool = False) -> Self
__init__(M: Int, M_end: Int, N: Int, K: Int, mma_shape: IndexList[3], cta_group: Int, cluster_shape: IndexList[3], block_swizzle_size: Int, rasterize_order: RasterOrder, swapAB: Bool = False, k_group_size: Int = 1, num_accum_pipeline_stages: Int = 2, num_clc_pipeline_stages: Int = 2, num_split_k: Int = 1, is_small_bn: Bool = False) -> 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!