From 00a74a2a886aac77f94faa55aff189a6e3b72e46 Mon Sep 17 00:00:00 2001
From: rolandschulz <roland.schulz@intel.com>
Date: Thu, 11 Apr 2024 21:03:35 -0700
Subject: [PATCH] Apply suggestions from code review

Co-authored-by: Mehdi Goli <mehdi.goli@codeplay.com>
---
 include/cute/arch/mma_xe.hpp         | 2 +-
 include/cute/atom/copy_atom.hpp      | 2 ++
 include/cute/atom/copy_traits_xe.hpp | 2 +-
 include/cute/atom/mma_atom.hpp       | 2 ++
 include/cute/atom/mma_traits_xe.hpp  | 2 +-
 include/cute/util/sycl_vec.hpp       | 2 +-
 6 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/include/cute/arch/mma_xe.hpp b/include/cute/arch/mma_xe.hpp
index f523d550f3..e0a9e27a3c 100644
--- a/include/cute/arch/mma_xe.hpp
+++ b/include/cute/arch/mma_xe.hpp
@@ -53,4 +53,4 @@ struct XE_1x16x16_BF16BF16F32F32_NN
     d = intel_sub_group_bf16_bf16_matrix_mad_k16(a, b, c);
   }
 };
-} //namespace cute
\ No newline at end of file
+} //namespace cute
diff --git a/include/cute/atom/copy_atom.hpp b/include/cute/atom/copy_atom.hpp
index 26b5aa798d..8431bc5e67 100644
--- a/include/cute/atom/copy_atom.hpp
+++ b/include/cute/atom/copy_atom.hpp
@@ -770,6 +770,8 @@ print_latex_copy(LayoutS const& S, ThrIDS const& TS,  // (m,n) -> (tid,vid)  and
 #include <cute/atom/copy_traits_sm90_tma.hpp>
 #endif
 
+#if defined(CUTLASS_ENABLE_SYCL)
 #include <cute/atom/copy_traits_xe.hpp>
+#endif
 
 ////////////////////////////////////////////////////////////////////////////////////////////////////
diff --git a/include/cute/atom/copy_traits_xe.hpp b/include/cute/atom/copy_traits_xe.hpp
index e0e13c5f45..b4023c0b40 100644
--- a/include/cute/atom/copy_traits_xe.hpp
+++ b/include/cute/atom/copy_traits_xe.hpp
@@ -60,4 +60,4 @@ namespace cute
         Traits traits{gtensor};
         return Copy_Atom<Traits, typename GEngine::value_type>{traits};
     }
-}
\ No newline at end of file
+}
diff --git a/include/cute/atom/mma_atom.hpp b/include/cute/atom/mma_atom.hpp
index 2c17752901..b1f0545b9c 100644
--- a/include/cute/atom/mma_atom.hpp
+++ b/include/cute/atom/mma_atom.hpp
@@ -938,5 +938,7 @@ print_latex_mma(LayoutC const& C, ThrIDC const& TC,  // (m,n) -> (tid,vid)  and
 #include <cute/atom/mma_traits_sm80.hpp>
 #include <cute/atom/mma_traits_sm90.hpp>
 #include <cute/atom/mma_traits_sm90_gmma.hpp>
+#if defined(CUTLASS_ENABLE_SYCL)
 #include <cute/atom/mma_traits_xe.hpp>
+#endif
 ////////////////////////////////////////////////////////////////////////////////////////////////////
diff --git a/include/cute/atom/mma_traits_xe.hpp b/include/cute/atom/mma_traits_xe.hpp
index 423ee48cf9..d90389be99 100644
--- a/include/cute/atom/mma_traits_xe.hpp
+++ b/include/cute/atom/mma_traits_xe.hpp
@@ -21,4 +21,4 @@ struct MMA_Traits<XE_8x16x16_BF16BF16F32F32_NN>
   using BLayout = Layout<Shape<_16, _16>, Stride<_16, _1>>;
   using CLayout = Layout<Shape<_8, _16>, Stride<_8, _1>>;
 };
-}
\ No newline at end of file
+}
diff --git a/include/cute/util/sycl_vec.hpp b/include/cute/util/sycl_vec.hpp
index 8202cc2a84..7c38d9c83a 100644
--- a/include/cute/util/sycl_vec.hpp
+++ b/include/cute/util/sycl_vec.hpp
@@ -14,4 +14,4 @@ using short8 = vector_t<short, 8>;
 using ushort8 = vector_t<ushort, 8>;
 using int2_ = vector_t<int, 2>; //conflicts with vector_types
 using int8 = vector_t<int, 8>;
-using uint8 = vector_t<uint, 8>;
\ No newline at end of file
+using uint8 = vector_t<uint, 8>;