diff --git a/include/cute/atom/mma_traits_xe.hpp b/include/cute/atom/mma_traits_xe.hpp index b70263fe00..423ee48cf9 100644 --- a/include/cute/atom/mma_traits_xe.hpp +++ b/include/cute/atom/mma_traits_xe.hpp @@ -10,10 +10,10 @@ namespace cute template <> struct MMA_Traits { - using ElementDVal = float; - using ElementAVal = sycl::ext::oneapi::bfloat16; - using ElementBVal = sycl::ext::oneapi::bfloat16; - using ElementCVal = float; + using ValTypeD = float; + using ValTypeA = sycl::ext::oneapi::bfloat16; + using ValTypeB = sycl::ext::oneapi::bfloat16; + using ValTypeC = float; using Shape_MNK = Shape<_8,_16,_16>; using ThrID = Layout<_16>;