Skip to main content

Module matmul

Module matmul 

Source
Expand description

The matmul reading of a Tile: c.mma(a, b) treats the trailing two axes as the row × col matrix, leading axes as a batch, and contracts K.

Structs§

MmaIo
Which transports a manual-mma tile may load through; every unit stores its own cells, since stmatrix does not reach a tile’s memory window. LoadMatrix lets an operand load through ldmatrix wherever the device offers it for the element and the stage serves it, which the tile decides as it expands; Manual keeps every unit loading its own cells. Default is what a caller with no reason to choose takes: ldmatrix for both operands where it serves.

Enums§

LoadMethod