cutlass icon indicating copy to clipboard operation
cutlass copied to clipboard

[QST] Does CUTE for SM90 support BFloat16 MMA?

Open ghostplant opened this issue 1 year ago • 0 comments

Does CUTE for SM90 support BFloat16 MMA?

I am searching for a MMA instruction designed for B16B16B16 from https://github.com/NVIDIA/cutlass/blob/main/include/cute/atom/mma_traits_sm90_gmma.hpp#L2824.

But the whole candidates don't have any Fully BF16BF16BF16 choice.

ghostplant avatar Oct 19 '24 20:10 ghostplant