include/ck_tile/ops/gemm.hpp Source File# Composable Kernel: include/ck_tile/ops/gemm.hpp Source File includeck_tileops gemm.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/gemm/block/block_gemm_areg_bgmem_creg_v1.hpp" 7 #include "ck_tile/ops/gemm/block/block_gemm_areg_bgmem_creg_v1_default_policy.hpp" 8 #include "ck_tile/ops/gemm/block/block_gemm_areg_breg_creg_v1.hpp" 9 #include "ck_tile/ops/gemm/block/block_gemm_areg_breg_creg_v1_custom_policy.hpp" 10 #include "ck_tile/ops/gemm/block/block_gemm_areg_breg_creg_v1_default_policy.hpp" 11 #include "ck_tile/ops/gemm/block/block_gemm_areg_bsmem_creg_one_warp_v1.hpp" 12 #include "ck_tile/ops/gemm/block/block_gemm_areg_bsmem_creg_v1.hpp" 13 #include "ck_tile/ops/gemm/block/block_gemm_areg_bsmem_creg_v1_custom_policy.hpp" 14 #include "ck_tile/ops/gemm/block/block_gemm_areg_bsmem_creg_v1_default_policy.hpp" 15 #include "ck_tile/ops/gemm/block/block_gemm_areg_bsmem_creg_v2.hpp" 16 #include "ck_tile/ops/gemm/block/block_gemm_areg_bsmem_creg_v2_custom_policy.hpp" 17 #include "ck_tile/ops/gemm/block/block_gemm_areg_bsmem_creg_v2_default_policy.hpp" 18 #include "ck_tile/ops/gemm/block/block_gemm_asmem_breg_creg_v1.hpp" 19 #include "ck_tile/ops/gemm/block/block_gemm_asmem_breg_creg_v1_custom_policy.hpp" 20 #include "ck_tile/ops/gemm/block/block_gemm_asmem_breg_creg_v1_default_policy.hpp" 21 #include "ck_tile/ops/gemm/block/block_gemm_asmem_bsmem_creg_v1.hpp" 22 #include "ck_tile/ops/gemm/block/block_gemm_asmem_bsmem_creg_v1_custom_policy.hpp" 23 #include "ck_tile/ops/gemm/block/block_gemm_asmem_bsmem_creg_v1_default_policy.hpp" 24 #include "ck_tile/ops/gemm/block/block_gemm_problem.hpp" 25 #include "ck_tile/ops/gemm/block/block_universal_gemm_as_bs_cr.hpp" 26 #include "ck_tile/ops/gemm/kernel/batched_gemm_kernel.hpp" 27 #include "ck_tile/ops/gemm/kernel/gemm_kernel.hpp" 28 #include "ck_tile/ops/gemm/kernel/gemm_tile_partitioner.hpp" 29 #include "ck_tile/ops/gemm/kernel/grouped_gemm_kernel.hpp" 30 #include "ck_tile/ops/gemm/pipeline/gemm_pipeline_ag_bg_cr_base.hpp" 31 #include "ck_tile/ops/gemm/pipeline/gemm_pipeline_ag_bg_cr_comp_v3.hpp" 32 #include "ck_tile/ops/gemm/pipeline/gemm_pipeline_ag_bg_cr_mem.hpp" 33 #include "ck_tile/ops/gemm/pipeline/gemm_pipeline_ag_bg_cr_scheduler.hpp" 34 #include "ck_tile/ops/gemm/pipeline/gemm_pipeline_agmem_bgmem_creg_v1.hpp" 35 #include "ck_tile/ops/gemm/pipeline/gemm_pipeline_agmem_bgmem_creg_v1_default_policy.hpp" 36 #include "ck_tile/ops/gemm/pipeline/gemm_pipeline_agmem_bgmem_creg_v2.hpp" 37 #include "ck_tile/ops/gemm/pipeline/gemm_pipeline_agmem_bgmem_creg_v2_default_policy.hpp" 38 #include "ck_tile/ops/gemm/pipeline/gemm_pipeline_problem.hpp" 39 #include "ck_tile/ops/gemm/pipeline/gemm_universal_pipeline_ag_bg_cr_policy.hpp" 40 #include "ck_tile/ops/gemm/pipeline/tile_gemm_shape.hpp" 41 #include "ck_tile/ops/gemm/pipeline/tile_gemm_traits.hpp" 42 #include "ck_tile/ops/gemm/warp/warp_gemm.hpp" 43 #include "ck_tile/ops/gemm/warp/warp_gemm_attribute_mfma.hpp" 44 #include "ck_tile/ops/gemm/warp/warp_gemm_attribute_mfma_impl.hpp" 45 #include "ck_tile/ops/gemm/warp/warp_gemm_dispatcher.hpp" 46 #include "ck_tile/ops/gemm/warp/warp_gemm_impl.hpp" 47 #include "ck_tile/ops/common/generic_2d_block_shape.hpp" 48 #include "ck_tile/ops/common/tensor_layout.hpp" batched_gemm_kernel.hpp block_gemm_areg_bgmem_creg_v1.hpp block_gemm_areg_bgmem_creg_v1_default_policy.hpp block_gemm_areg_breg_creg_v1.hpp block_gemm_areg_breg_creg_v1_custom_policy.hpp block_gemm_areg_breg_creg_v1_default_policy.hpp block_gemm_areg_bsmem_creg_one_warp_v1.hpp block_gemm_areg_bsmem_creg_v1.hpp block_gemm_areg_bsmem_creg_v1_custom_policy.hpp block_gemm_areg_bsmem_creg_v1_default_policy.hpp block_gemm_areg_bsmem_creg_v2.hpp block_gemm_areg_bsmem_creg_v2_custom_policy.hpp block_gemm_areg_bsmem_creg_v2_default_policy.hpp block_gemm_asmem_breg_creg_v1.hpp block_gemm_asmem_breg_creg_v1_custom_policy.hpp block_gemm_asmem_breg_creg_v1_default_policy.hpp block_gemm_asmem_bsmem_creg_v1.hpp block_gemm_asmem_bsmem_creg_v1_custom_policy.hpp block_gemm_asmem_bsmem_creg_v1_default_policy.hpp block_gemm_problem.hpp block_universal_gemm_as_bs_cr.hpp gemm_kernel.hpp gemm_pipeline_ag_bg_cr_base.hpp gemm_pipeline_ag_bg_cr_comp_v3.hpp gemm_pipeline_ag_bg_cr_mem.hpp gemm_pipeline_ag_bg_cr_scheduler.hpp gemm_pipeline_agmem_bgmem_creg_v1.hpp gemm_pipeline_agmem_bgmem_creg_v1_default_policy.hpp gemm_pipeline_agmem_bgmem_creg_v2.hpp gemm_pipeline_agmem_bgmem_creg_v2_default_policy.hpp gemm_pipeline_problem.hpp gemm_tile_partitioner.hpp gemm_universal_pipeline_ag_bg_cr_policy.hpp generic_2d_block_shape.hpp grouped_gemm_kernel.hpp tensor_layout.hpp tile_gemm_shape.hpp tile_gemm_traits.hpp warp_gemm.hpp warp_gemm_attribute_mfma.hpp warp_gemm_attribute_mfma_impl.hpp warp_gemm_dispatcher.hpp warp_gemm_impl.hpp