ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-29 19:28:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

made changes to cshuffle datatype to resolve compilation error: reduction is also working for bf16, fp8

Browse Source
This commit is contained in:
Astha Rai
2025-06-04 20:42:32 +00:00
parent 5f9d447669
commit 64451ec381
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
example/01_gemm/gemm_xdl_bf16_streamk_v3.cpp
View File

@@ -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;

2
example/01_gemm/gemm_xdl_fp8_streamk_v3.cpp
View File

@@ -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;