flatmm.hpp Source File# Composable Kernel: flatmm.hpp Source File includeck_tileops flatmm.hpp Go to the documentation of this file. 1// SPDX-License-Identifier: MIT 2// Copyright (c) 2018-2025, Advanced Micro Devices, Inc. All rights reserved. 3 4#pragma once 5 6#include "ck_tile/ops/flatmm/block/block_flatmm_asmem_bsmem_creg_v1.hpp" 7#include "ck_tile/ops/flatmm/block/block_flatmm_asmem_bsmem_creg_v1_custom_policy.hpp" 8#include "ck_tile/ops/flatmm/block/flatmm_32x512x128_1x4x1_16x16x32.hpp" 9#include "ck_tile/ops/flatmm/block/flatmm_sn_32x128x512_1x4x1_16x16x32.hpp" 10#include "ck_tile/ops/flatmm/block/flatmm_sn_32x128x512_1x4x1_16x16x32_itl.hpp" 11#include "ck_tile/ops/flatmm/block/flatmm_uk_config.hpp" 12#include "ck_tile/ops/flatmm/kernel/flatmm_kernel.hpp" 13#include "ck_tile/ops/flatmm/kernel/grouped_flatmm_kernel.hpp" 14#include "ck_tile/ops/flatmm/kernel/mixed_prec_flatmm_kernel.hpp" 15#include "ck_tile/ops/flatmm/kernel/moe_flatmm_kernel.hpp" 16#include "ck_tile/ops/flatmm/kernel/mx_flatmm_kernel.hpp" 17#include "ck_tile/ops/flatmm/pipeline/flatmm_pipeline_agmem_bgmem_creg_v1.hpp" 18#include "ck_tile/ops/flatmm/pipeline/flatmm_pipeline_agmem_bgmem_creg_v1_policy.hpp" 19#include "ck_tile/ops/flatmm/pipeline/mixed_prec_flatmm_pipeline_agmem_bgmem_creg_v1.hpp" 20#include "ck_tile/ops/flatmm/pipeline/mixed_prec_flatmm_pipeline_agmem_bgmem_creg_v1_policy.hpp" 21#include "ck_tile/ops/flatmm/pipeline/moe_flatmm_pipeline_agmem_bgmem_creg.hpp" 22#include "ck_tile/ops/flatmm/pipeline/mx_flatmm_pipeline_agmem_bgmem_creg_v1.hpp" 23#include "ck_tile/ops/flatmm/pipeline/mx_flatmm_pipeline_agmem_bgmem_creg_v1_policy.hpp" 24#include "ck_tile/ops/flatmm/pipeline/tile_flatmm_shape.hpp" 25#include "ck_tile/ops/common/generic_2d_block_shape.hpp" 26#include "ck_tile/ops/common/load_interleaved_pk_type.hpp" 27#include "ck_tile/ops/common/streamk_common.hpp" 28#include "ck_tile/ops/common/tensor_layout.hpp" 29#include "ck_tile/ops/common/utils.hpp" block_flatmm_asmem_bsmem_creg_v1.hpp block_flatmm_asmem_bsmem_creg_v1_custom_policy.hpp flatmm_32x512x128_1x4x1_16x16x32.hpp flatmm_kernel.hpp flatmm_pipeline_agmem_bgmem_creg_v1.hpp flatmm_pipeline_agmem_bgmem_creg_v1_policy.hpp flatmm_sn_32x128x512_1x4x1_16x16x32.hpp flatmm_sn_32x128x512_1x4x1_16x16x32_itl.hpp flatmm_uk_config.hpp generic_2d_block_shape.hpp grouped_flatmm_kernel.hpp load_interleaved_pk_type.hpp mixed_prec_flatmm_kernel.hpp mixed_prec_flatmm_pipeline_agmem_bgmem_creg_v1.hpp mixed_prec_flatmm_pipeline_agmem_bgmem_creg_v1_policy.hpp moe_flatmm_kernel.hpp moe_flatmm_pipeline_agmem_bgmem_creg.hpp mx_flatmm_kernel.hpp mx_flatmm_pipeline_agmem_bgmem_creg_v1.hpp mx_flatmm_pipeline_agmem_bgmem_creg_v1_policy.hpp streamk_common.hpp tensor_layout.hpp tile_flatmm_shape.hpp utils.hpp