Mojo module
amd_ping_pong_matmul
Structured ping-pong matmul for AMD MI355X (CDNA4).
Entry point: AMDPingPongMatmul.run() Host launcher: structured_ping_pong_matmul()
Structsβ
- β
AMDPingPongMatmul: Structured ping-pong matmul for AMD MI355X. - β
KernelConfig: Block/warp/MMA shape configuration for ping-pong kernels.
Functionsβ
- β
s_barrier: - β
s_setprio: - β
structured_ping_pong_matmul: Host launcher for the structured ping-pong matmul.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!