Alexander Efimov 6691de65db [MFMA] Support BFloat16 on MI100 (#295) ...

* [MFMA] Support BFloat16 on MI100

This PR makes use of mfma_f32_32x32x4bf16 instruction, available on MI100.

* fix tests, fix mfma encoding comment, fix switch between mfma versions.

* replace kDim from mfma layout with kWidth from dotOp layout

* rebase fix

* fix mfma to dot op shortcut for bfloat16

* fix review comments

2023-09-08 15:08:34 -05:00

21 KiB

Raw Blame History

View Raw