Mojo module
testbed_swapAB
Testbed for comparing swapAB vs normal matmul execution.
swapAB is an internal optimization where:
- A and B operands are swapped inside the kernel
- The output C tile is transposed on write-out
- The final result C[M,N] should be identical to the normal kernel
Both kernels compute: C[M,N] = A[M,K] @ B[N,K]^T The swapAB version just does it via: (B @ A^T)^T stored transposed = A @ B^T
comptime values
UnsafePointer
comptime UnsafePointer = LegacyUnsafePointer[?, address_space=?, origin=?]
Functions
-
test_matmul_sm90_swapAB_comparison: Compare matmul results between normal execution and swapAB execution. -
test_matmul_sm90_swapAB_comparison_v2: Compare matmul results between normal execution and swapAB execution.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!