For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /max/get-started.md).

Mojo function

test_matmul_sm90_swapAB_comparison

def test_matmul_sm90_swapAB_comparison[a_type: DType, b_type: DType, c_type: DType, config: MatmulConfig[a_type, b_type, c_type], config_swapAB: MatmulConfig[a_type, b_type, c_type], MType: CoordLike, NType: CoordLike, KType: CoordLike, //](ctx: DeviceContext, m: MType, n: NType, k: KType)

Compare matmul results between normal execution and swapAB execution.

Both compute: C[M,N] = A[M,K] @ B[N,K]^T swapAB internally swaps A/B and transposes C on store, but result should match.

Args:

ctx (DeviceContext): The device context.
m (MType): The M dimension (can be static or dynamic).
n (NType): The N dimension (can be static or dynamic).
k (KType): The K dimension (can be static or dynamic).