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;