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

update scale-preshuffle for MXF4

Browse Source
This commit is contained in:
Feng Shijie
2025-08-13 10:48:53 +00:00
parent edb58d0680
commit 732ebdee8b
6 changed files with 376 additions and 401 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
include/ck_tile/ops/flatmm/kernel/mixed_prec_flatmm_kernel.hpp
View File

@@ -13,11 +13,8 @@
namespace ck_tile {
template <typename TilePartitioner_,
typename FlatmmPipeline_,
typename EpiloguePipeline_,
int SupportArch = 0> // 0 means no arch restriction
struct MixedPrecFlatmmKernel : FlatmmKernel<TilePartitioner_, FlatmmPipeline_, EpiloguePipeline_>
template <typename TilePartitioner_, typename FlatmmPipeline_, typename EpiloguePipeline_>
struct F16xMXF4FlatmmKernel : FlatmmKernel<TilePartitioner_, FlatmmPipeline_, EpiloguePipeline_>
{
using Underlying = FlatmmKernel<TilePartitioner_, FlatmmPipeline_, EpiloguePipeline_>;