From 64451ec38176a0bb5f20eb89e4007764f360a0df Mon Sep 17 00:00:00 2001
From: Astha Rai <astha.rai713@gmail.com>
Date: Wed, 4 Jun 2025 20:42:32 +0000
Subject: [PATCH] made changes to cshuffle datatype to resolve compilation
 error: reduction is also working for bf16, fp8

---
 example/01_gemm/gemm_xdl_bf16_streamk_v3.cpp | 2 +-
 example/01_gemm/gemm_xdl_fp8_streamk_v3.cpp  | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/example/01_gemm/gemm_xdl_bf16_streamk_v3.cpp b/example/01_gemm/gemm_xdl_bf16_streamk_v3.cpp
index 5b56a43483..98125eddfa 100755
--- a/example/01_gemm/gemm_xdl_bf16_streamk_v3.cpp
+++ b/example/01_gemm/gemm_xdl_bf16_streamk_v3.cpp
@@ -9,7 +9,7 @@ using ADataType        = ck::bhalf_t;
 using BDataType        = ck::bhalf_t;
 using CDataType        = ck::bhalf_t;
 using AccDataType      = float;
-using CShuffleDataType = ck::bhalf_t;
+using CShuffleDataType = float;
 
 using ALayout = Row;
 using BLayout = Col;
diff --git a/example/01_gemm/gemm_xdl_fp8_streamk_v3.cpp b/example/01_gemm/gemm_xdl_fp8_streamk_v3.cpp
index 3b79ae9b85..6c7fbb5481 100755
--- a/example/01_gemm/gemm_xdl_fp8_streamk_v3.cpp
+++ b/example/01_gemm/gemm_xdl_fp8_streamk_v3.cpp
@@ -8,7 +8,7 @@
 using ADataType        = ck::f8_t;
 using BDataType        = ck::f8_t;
 using AccDataType      = float;
-using CShuffleDataType = ck::half_t;
+using CShuffleDataType = float;
 using CDataType        = ck::half_t;
 
 using ALayout = Row;