Added UTs for different datatype in epilogue

Signed-off-by: Chawla, Amit K <[email protected]>

Author: amitchawla1

include/cutlass/epilogue/collective/xe_array_epilogue.hpp

@@ -132,19 +132,22 @@ class CollectiveEpilogue<
   using CopyThreadShape = Shape<_1, Int<SubgroupSize>>;
 
   using Trait_D = Copy_Traits<GmemTiledCopyD, InternalStrideD>;
+  using val_layout_store_D = decltype(make_layout(shape_div(typename Trait_D::BlockShape{}, CopyThreadShape{})));
   using XE_Copy_D = decltype(make_tiled_copy(Copy_Atom<Trait_D, ElementD>{},
                                              Layout<CopyThreadShape>{},
-                                             make_layout(shape_div(typename Trait_D::BlockShape{}, CopyThreadShape{}))));
+                                             val_layout_store_D{}));
 private:
   constexpr static bool is_source_supported = not cute::is_void_v<ElementC>;
   constexpr static bool is_destination_supported = not cute::is_void_v<ElementD> && not cute::is_void_v<CopyOpR2G>;
 
   using NonVoidElementC = conditional_t<is_source_supported, ElementC, ElementD>;
   using Trait_C = Copy_Traits<GmemTiledCopyC, InternalStrideC>;
   using NonVoidTrait_C = conditional_t<is_source_supported, Trait_C, Trait_D>;
+  using val_layout_load_C = decltype(make_layout(shape_div(typename NonVoidTrait_C::BlockShape{}, CopyThreadShape{})));
+  using NonVoidValLayoutLoad_C = conditional_t<is_source_supported, val_layout_load_C, val_layout_store_D>;
   using XE_Copy_C = decltype(make_tiled_copy(Copy_Atom<NonVoidTrait_C, NonVoidElementC>{},
                                              Layout<CopyThreadShape>{},
-                                             make_layout(shape_div(typename NonVoidTrait_C::BlockShape{}, CopyThreadShape{}))));
+                                             NonVoidValLayoutLoad_C{}));
 public:
 
   using EmptyType = cute::tuple<>;

test/unit/gemm/device/xe_gemm_bf16_bf16_fp32_tensor_op_bf16.cpp

@@ -1,5 +1,5 @@
 /***************************************************************************************************
- * Copyright (c) 2025 - 2025 Codeplay Software Ltd. All rights reserved.
+ * Copyright (C) 2025 Intel Corporation, All rights reserved.
  * SPDX-License-Identifier: BSD-3-Clause
  *
  * Redistribution and use in source and binary forms, with or without
@@ -27,7 +27,7 @@
  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  *
- **************************************************************************************************/
+ ***************************************************************************************************/
 
 /*! \file
     \brief Tests for Xe bf16_bf16_fp32 and C is bf16