ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-19 12:30:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix for Unsupported Input Shapes/Sizes in Stream-K GEMM - BF16/FP16 (#1866)

Browse Source 
[ROCm/composable_kernel commit: 92b79ead0a]
This commit is contained in:
Muhammed Emin Ozturk
2025-02-18 08:46:47 -08:00
committed by GitHub
parent 16162174fd
commit baeffc2345
5 changed files with 85 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
example/67_gemm_microscaling/gemm_mx_common.hpp
View File

@@ -13,7 +13,9 @@
#include "ck/utility/blkgemmpipe_scheduler.hpp"
#include "ck/utility/data_type.hpp"
#include "ck/utility/sequence.hpp"
#include "ck/library/reference_tensor_operation/cpu/reference_mx_gemm.hpp"
#include "ck/library/utility/check_err.hpp"
#include "ck/library/utility/device_memory.hpp"
#include "ck/library/utility/fill.hpp"