Update to 3.5 API

AD2605 · May 24, 2024 · 2773705 · 2773705
1 parent 63912b1
commit 2773705
Showing 1 changed file with 4 additions and 4 deletions.
diff --git 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<XE_8x16x16_BF16BF16F32F32_NN>
 {
-  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>;