Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
gemm.hpp 2.47 KB
Newer Older
carlushuang's avatar
introducing ck_tile! (#1216)  
carlushuang committed
1
2
3
4
5
6
7 
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-2024, Advanced Micro Devices, Inc. All rights reserved.

#pragma once

#include "ck_tile/ops/gemm/block/block_gemm_areg_bgmem_creg_v1.hpp"
#include "ck_tile/ops/gemm/block/block_gemm_areg_bgmem_creg_v1_default_policy.hpp"
Dan Yao's avatar
[CK_TILE] FA bwd kernels optimization (#1397)  
Dan Yao committed
8
9
10 
#include "ck_tile/ops/gemm/block/block_gemm_areg_breg_creg_v1.hpp"
#include "ck_tile/ops/gemm/block/block_gemm_areg_breg_creg_v1_custom_policy.hpp"
#include "ck_tile/ops/gemm/block/block_gemm_areg_breg_creg_v1_default_policy.hpp"
carlushuang's avatar
introducing ck_tile! (#1216)  
carlushuang committed
11
12
13
14
15
16 
#include "ck_tile/ops/gemm/block/block_gemm_areg_bsmem_creg_v1.hpp"
#include "ck_tile/ops/gemm/block/block_gemm_areg_bsmem_creg_v1_custom_policy.hpp"
#include "ck_tile/ops/gemm/block/block_gemm_areg_bsmem_creg_v1_default_policy.hpp"
#include "ck_tile/ops/gemm/block/block_gemm_areg_bsmem_creg_v2.hpp"
#include "ck_tile/ops/gemm/block/block_gemm_areg_bsmem_creg_v2_custom_policy.hpp"
#include "ck_tile/ops/gemm/block/block_gemm_areg_bsmem_creg_v2_default_policy.hpp"
Dan Yao's avatar
CK Tile FA Training kernels (#1286)  
Dan Yao committed
17
18
19 
#include "ck_tile/ops/gemm/block/block_gemm_asmem_breg_creg_v1.hpp"
#include "ck_tile/ops/gemm/block/block_gemm_asmem_breg_creg_v1_custom_policy.hpp"
#include "ck_tile/ops/gemm/block/block_gemm_asmem_breg_creg_v1_default_policy.hpp"
carlushuang's avatar
introducing ck_tile! (#1216)  
carlushuang committed
20
21
22 
#include "ck_tile/ops/gemm/block/block_gemm_asmem_bsmem_creg_v1.hpp"
#include "ck_tile/ops/gemm/block/block_gemm_asmem_bsmem_creg_v1_custom_policy.hpp"
#include "ck_tile/ops/gemm/block/block_gemm_asmem_bsmem_creg_v1_default_policy.hpp"
Dan Yao's avatar
CK Tile FA Training kernels (#1286)  
Dan Yao committed
23 
#include "ck_tile/ops/gemm/block/block_gemm_problem.hpp"
Thomas Ning's avatar
Ck tile gemm example (#1488)  
Thomas Ning committed
24
25 
#include "ck_tile/ops/gemm/kernel/gemm_kernel.hpp"
#include "ck_tile/ops/gemm/kernel/gemm_tile_partitioner.hpp"
Bartłomiej Kocot's avatar
[CK_TILE] Add block universal gemm pipeline policy (#1557)  
Bartłomiej Kocot committed
26 
#include "ck_tile/ops/gemm/pipeline/gemm_universal_pipeline_ag_bg_cr_policy.hpp"
Thomas Ning's avatar
Ck tile gemm cshuffle & CK Tile GEMM restructure (#1535)  
Thomas Ning committed
27
28
29
30
31 
#include "ck_tile/ops/gemm/pipeline/gemm_pipeline_agmem_bgmem_creg_v1.hpp"
#include "ck_tile/ops/gemm/pipeline/gemm_pipeline_agmem_bgmem_creg_v1_default_policy.hpp"
#include "ck_tile/ops/gemm/pipeline/gemm_pipeline_agmem_bgmem_creg_v2.hpp"
#include "ck_tile/ops/gemm/pipeline/gemm_pipeline_agmem_bgmem_creg_v2_default_policy.hpp"
#include "ck_tile/ops/gemm/pipeline/gemm_pipeline_problem.hpp"
carlushuang's avatar
introducing ck_tile! (#1216)  
carlushuang committed
32 
#include "ck_tile/ops/gemm/pipeline/tile_gemm_shape.hpp"
Thomas Ning's avatar
Ck tile gemm cshuffle & CK Tile GEMM restructure (#1535)  
Thomas Ning committed
33 
#include "ck_tile/ops/gemm/pipeline/tile_gemm_traits.hpp"
carlushuang's avatar
introducing ck_tile! (#1216)  
carlushuang committed
34
35
36
37
38
39 
#include "ck_tile/ops/gemm/warp/warp_gemm.hpp"
#include "ck_tile/ops/gemm/warp/warp_gemm_attribute_mfma.hpp"
#include "ck_tile/ops/gemm/warp/warp_gemm_attribute_mfma_impl.hpp"
#include "ck_tile/ops/gemm/warp/warp_gemm_dispatcher.hpp"
#include "ck_tile/ops/gemm/warp/warp_gemm_impl.hpp"
#include "ck_tile/ops/common/tensor_layout.hpp"