๊ฐ€์ž… ํ›„ ์ดˆ๋Œ€ ๋งํฌ๋ฅผ ๊ณต์œ ํ•˜๋ฉด ๋™์˜์ƒ ์žฌ์ƒ ๋ฐ ์ดˆ๋Œ€ ๋ณด์ƒ์„ ๋ฐ›์„ ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.

Thien Tran
@gaunernst
๊ฐ€์ž… October 2016
276 ํŒ”๋กœ์ž‰ ์ค‘    3.1K ํŒฌ
Thank you for everyone who commented on other public MXFP8 kernels that I didn't know about. At least for the toy 8192^3 benchmark shape here, K3-generated Gluon is the best ๐Ÿ˜ณ I haven't checked all the kernels closely but I'm sure they can shine under different GEMM shapes + tuning knobs. Having an assemble of kernels is useful for autotuning - it is effortless thanks to agents now.
๋” ๋ณด๊ธฐ
I couldn't find a public performant MXFP8 GEMM on MI355X, so I just asked @KimiDevs K3 to write one, and it beats torch._scaled_mm by >2x times ๐Ÿคฏ This is an adaptation of the official Gluon MXFP4 example. FlyDSL adaptation is WIP. @AnushElangovan you guys should try K3 for kernel engineering if you haven't!
๋” ๋ณด๊ธฐ