Mojo function

test_matmul_sm90_swapAB_comparison

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]](ctx: DeviceContext, m: ValOrDim[dim], n: ValOrDim[dim], k: ValOrDim[dim])

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 (ValOrDim): The M dimension (can be static or dynamic).
n (ValOrDim): The N dimension (can be static or dynamic).
k (ValOrDim): The K dimension (can be static or dynamic).