ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-04-20 06:49:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

[CK_TILE] add generic_permute (#1607)

Browse Source
This commit is contained in:
valarLip
2024-10-29 18:05:53 +08:00
committed by GitHub
parent 922e42a039
commit 9fbd72e97e
14 changed files with 1318 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
include/ck_tile/host.hpp
View File

@@ -22,6 +22,7 @@
#include "ck_tile/host/reference/reference_gemm.hpp"
#include "ck_tile/host/reference/reference_im2col.hpp"
#include "ck_tile/host/reference/reference_layernorm2d_fwd.hpp"
#include "ck_tile/host/reference/reference_permute.hpp"
#include "ck_tile/host/reference/reference_reduce.hpp"
#include "ck_tile/host/reference/reference_softmax.hpp"
#include "ck_tile/host/reference/reference_topk.hpp"