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

MFMA 16x16x32fp8 (#2103)

Browse Source 
* add mfma_16x16x32_fp8

* clang format code

* Finished the fix for gemm basic

* clang foramt

* rebuild CI

* recover gemm.hpp

* add MFMA 16*16*32bf8

---------

Co-authored-by: solin <bingzhou@amd.com>
This commit is contained in:
Thomas Ning
2025-04-21 10:21:35 -07:00
committed by GitHub
parent ce61759538
commit a738e43445
5 changed files with 188 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
include/ck_tile/ops/gemm/pipeline/gemm_pipeline_problem.hpp
View File

@@ -32,6 +32,8 @@ struct GemmPipelineProblemBase
static constexpr bool TransposeC = Traits::TransposeC;
static constexpr bool UseStructuredSparsity = Traits::UseStructuredSparsity;
static constexpr index_t kBlockSize = BlockGemmShape::NumWarps * get_warp_size();
static constexpr bool kPadM = Traits::kPadM;

3
include/ck_tile/ops/gemm/pipeline/tile_gemm_traits.hpp
View File

@@ -26,7 +26,8 @@ struct TileGemmTraits
using BLayout = BLayout_;
using CLayout = CLayout_;
static constexpr bool TransposeC = false;
static constexpr bool TransposeC = false;
static constexpr bool UseStructuredSparsity = false;
};
template <bool kPadM_,