Skip to main content

Mojo function

mi355x_single_buffer_cost_model

mi355x_single_buffer_cost_model() -> TargetCostModel

MI355X cost model for single-buffer matmul (DefaultMatmulOps tags).

Tag mapping (from DefaultMatmulOps): 0=LOAD_DRAM: GLOBAL_MEM, 200 cycles, GLOBAL_LOAD 1=STORE_SMEM: LDS, 20 cycles, SHARED_STORE 2=LOAD_FRAG: LDS, 20 cycles, FRAGMENT_LOAD 3=COMPUTE: MMA_UNIT, 64 cycles, COMPUTE

Returns:

TargetCostModel

Was this page helpful?