fix XEHPC 2D load

Author: sunjiweiswift

include/experimental/group/gemm/impl/int4_dequantize_xe.hpp

@@ -210,7 +210,7 @@ class gemm_t<
               typename mem_desc_b_t::dtype,
               mem_layout::row_major,
               mem_desc_b_t::space>>,
-      // subgroup::msg_type_v<matB_tile_desc_t, mem_desc_b_t>,
+      //   subgroup::msg_type_v<matB_tile_desc_t, mem_desc_b_t>,
       arch_tag>;
   using matB_prefetch_payload_t = subgroup::
       prefetch_payload_t<mem_desc_b_t, matB_tile_desc_t, wg_size_y, arch_tag>;

include/subgroup/tile/impl/payload_xe.hpp

@@ -65,15 +65,14 @@ struct mem_payload_t<
       mem_payload_t<mem_desc_t, tile_desc, msg_type::block_2d, arch_tag>;
 
  public:
-  static constexpr bool mem_transpose =
-      memory_layout == mem_layout::col_major &&
-      !(std::is_same_v<dtype_, int4x2> || std::is_same_v<dtype_, int4x8>);
+  static constexpr bool mem_transpose = memory_layout == mem_layout::col_major;
 
   static constexpr reg_layout register_layout = tile_desc::register_layout;
   static constexpr bool reg_transpose =
       register_layout == reg_layout::transpose_tiled;
 
-  static constexpr bool trans = mem_transpose ^ reg_transpose;
+  static constexpr bool trans = (mem_transpose ^ reg_transpose) &&
+      !(std::is_same_v<dtype_, int4x2> || std::is_same_v<dtype_, int4x8>);
 
   static constexpr bool mem_transform = (sizeof(dtype) < 4) && !mem_transpose &&
       (register_layout == reg_layout::vnni_tiled ||
@@ -1094,7 +1093,7 @@ struct mem_payload_t<
   static constexpr reg_layout register_layout = tile_desc::register_layout;
   static constexpr bool reg_transpose =
       register_layout == reg_layout::transpose_tiled;
-  static constexpr bool trans = mem_transpose ^ reg_transpose &&
+  static constexpr bool trans = (mem_transpose ^ reg_transpose) &&
       !(std::is_same_v<dtype_, int4x2> || std::is_same_v<dtype_, int4x8>);
 
   static constexpr bool mem_transform = (sizeof(dtype) < 4) &&
@@ -1657,7 +1656,7 @@ struct prefetch_payload_t<
   static constexpr reg_layout register_layout = tile_desc::register_layout;
   static constexpr bool reg_transpose =
       register_layout == reg_layout::transpose_tiled;
-  static constexpr bool trans = mem_transpose ^ reg_transpose &&
+  static constexpr bool trans = (mem_transpose ^ reg_transpose) &&
       !(std::is_same_v<dtype_, int4x2> || std::is_same_v<dtype_, int4x8>);
 
   using prefetch_dtype = typename std::conditional<