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

[CK_TILE] Add 2:4 structured sparsity support for fp16 gemm (#1957)

Browse Source 
* add structured sparsity fp16 support for gemm

* added reviewer suggestions

* update changelog

* update changelog

* add reviewers suggestions

* Minor fix

* clang fix

* fix doxygen

[ROCm/composable_kernel commit: 6c61f4d237]
This commit is contained in:
jakpiase
2025-04-11 12:18:26 +02:00
committed by GitHub
parent cca9cca699
commit addcd203eb
13 changed files with 401 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
example/ck_tile/03_gemm/universal_gemm.cpp
View File

@@ -46,7 +46,8 @@ float gemm_calc(const ck_tile::GemmHostArgs& args, const ck_tile::stream_config&
ALayout,
BLayout,
CLayout,
GemmConfig::TransposeC>;
GemmConfig::TransposeC,
GemmConfig::UseStructuredSparsity>;
using GemmPipelineProblem =
ck_tile::GemmPipelineProblem<ADataType, BDataType, AccDataType, GemmShape, Traits>;