Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
gaoqiong
composable_kernel_ROCM
Commits
f8107760
Commit
f8107760
authored
Jan 19, 2025
by
Aleksander Dudek
Browse files
[CK_TILE] Update the file structure part2
parent
ec1b65e7
Changes
37
Show whitespace changes
Inline
Side-by-side
Showing
20 changed files
with
22 additions
and
22 deletions
+22
-22
example/ck_tile/04_img2col/image_to_column.hpp
example/ck_tile/04_img2col/image_to_column.hpp
+1
-1
example/ck_tile/06_permute/permute.hpp
example/ck_tile/06_permute/permute.hpp
+1
-1
example/ck_tile/09_topk_softmax/topk_softmax_api.hpp
example/ck_tile/09_topk_softmax/topk_softmax_api.hpp
+1
-1
example/ck_tile/10_rmsnorm2d/example_rmsnorm2d_fwd.cpp
example/ck_tile/10_rmsnorm2d/example_rmsnorm2d_fwd.cpp
+1
-1
example/ck_tile/10_rmsnorm2d/rmsnorm2d_fwd.hpp
example/ck_tile/10_rmsnorm2d/rmsnorm2d_fwd.hpp
+1
-1
example/ck_tile/11_add_rmsnorm2d_rdquant/add_rmsnorm2d_rdquant_fwd.hpp
...le/11_add_rmsnorm2d_rdquant/add_rmsnorm2d_rdquant_fwd.hpp
+1
-1
example/ck_tile/11_add_rmsnorm2d_rdquant/example_add_rmsnorm2d_rdquant_fwd.cpp
...d_rmsnorm2d_rdquant/example_add_rmsnorm2d_rdquant_fwd.cpp
+1
-1
example/ck_tile/12_smoothquant/example_smoothquant.cpp
example/ck_tile/12_smoothquant/example_smoothquant.cpp
+1
-1
example/ck_tile/12_smoothquant/smoothquant.hpp
example/ck_tile/12_smoothquant/smoothquant.hpp
+1
-1
example/ck_tile/13_moe_sorting/moe_sorting_api.hpp
example/ck_tile/13_moe_sorting/moe_sorting_api.hpp
+1
-1
example/ck_tile/14_moe_smoothquant/moe_smoothquant.hpp
example/ck_tile/14_moe_smoothquant/moe_smoothquant.hpp
+1
-1
example/ck_tile/15_fused_moe/fused_moegemm.hpp
example/ck_tile/15_fused_moe/fused_moegemm.hpp
+1
-1
example/ck_tile/15_fused_moe/fused_moesorting.hpp
example/ck_tile/15_fused_moe/fused_moesorting.hpp
+1
-1
example/ck_tile/15_fused_moe/instances/fused_moegemm_api_internal.hpp
...ile/15_fused_moe/instances/fused_moegemm_api_internal.hpp
+1
-1
include/ck_tile/device/ops/layernorm2d/pipeline/layernorm2d_fwd_pipeline_default_policy.hpp
...rm2d/pipeline/layernorm2d_fwd_pipeline_default_policy.hpp
+2
-2
include/ck_tile/host/ops/add_rmsnorm2d_rdquant/pipeline/add_rmsnorm2d_rdquant_fwd_pipeline_one_pass.hpp
.../pipeline/add_rmsnorm2d_rdquant_fwd_pipeline_one_pass.hpp
+1
-1
include/ck_tile/host/ops/add_rmsnorm2d_rdquant/pipeline/add_rmsnorm2d_rdquant_fwd_pipeline_three_pass.hpp
...ipeline/add_rmsnorm2d_rdquant_fwd_pipeline_three_pass.hpp
+1
-1
include/ck_tile/host/ops/flatmm/block/flatmm_32x512x128_1x4x1_16x16x32.hpp
...ost/ops/flatmm/block/flatmm_32x512x128_1x4x1_16x16x32.hpp
+1
-1
include/ck_tile/host/ops/flatmm/block/flatmm_sn_32x128x512_1x4x1_16x16x32.hpp
.../ops/flatmm/block/flatmm_sn_32x128x512_1x4x1_16x16x32.hpp
+1
-1
include/ck_tile/host/ops/flatmm/block/flatmm_sn_32x128x512_1x4x1_16x16x32_itl.hpp
.../flatmm/block/flatmm_sn_32x128x512_1x4x1_16x16x32_itl.hpp
+2
-2
No files found.
example/ck_tile/04_img2col/image_to_column.hpp
View file @
f8107760
...
@@ -5,7 +5,7 @@
...
@@ -5,7 +5,7 @@
#include "ck_tile/core.hpp"
#include "ck_tile/core.hpp"
#include "ck_tile/host/util/kernel_launch.hpp"
#include "ck_tile/host/util/kernel_launch.hpp"
#include "ck_tile/
device
/ops/image_to_column.hpp"
#include "ck_tile/
host
/ops/image_to_column.hpp"
#include <string>
#include <string>
#define DefaultConvParams \
#define DefaultConvParams \
...
...
example/ck_tile/06_permute/permute.hpp
View file @
f8107760
...
@@ -5,7 +5,7 @@
...
@@ -5,7 +5,7 @@
#include "ck_tile/core.hpp"
#include "ck_tile/core.hpp"
#include "ck_tile/host/util/kernel_launch.hpp"
#include "ck_tile/host/util/kernel_launch.hpp"
#include "ck_tile/
device
/ops/permute.hpp"
#include "ck_tile/
host
/ops/permute.hpp"
#include <string>
#include <string>
struct
permute_traits
struct
permute_traits
...
...
example/ck_tile/09_topk_softmax/topk_softmax_api.hpp
View file @
f8107760
...
@@ -4,7 +4,7 @@
...
@@ -4,7 +4,7 @@
#pragma once
#pragma once
#include "ck_tile/core.hpp"
#include "ck_tile/core.hpp"
#include "ck_tile/host.hpp"
#include "ck_tile/host.hpp"
#include "ck_tile/
device
/ops/topk_softmax.hpp"
#include "ck_tile/
host
/ops/topk_softmax.hpp"
#include <string>
#include <string>
struct
topk_softmax_trait
struct
topk_softmax_trait
...
...
example/ck_tile/10_rmsnorm2d/example_rmsnorm2d_fwd.cpp
View file @
f8107760
#include "ck_tile/host.hpp"
#include "ck_tile/host.hpp"
#include "ck_tile/core.hpp"
#include "ck_tile/core.hpp"
#include "ck_tile/host/util/kernel_launch.hpp"
#include "ck_tile/host/util/kernel_launch.hpp"
#include "ck_tile/
device
/ops/rmsnorm2d.hpp"
#include "ck_tile/
host
/ops/rmsnorm2d.hpp"
#include <cstring>
#include <cstring>
auto
create_args
(
int
argc
,
char
*
argv
[])
auto
create_args
(
int
argc
,
char
*
argv
[])
...
...
example/ck_tile/10_rmsnorm2d/rmsnorm2d_fwd.hpp
View file @
f8107760
...
@@ -5,7 +5,7 @@
...
@@ -5,7 +5,7 @@
#include "ck_tile/core.hpp"
#include "ck_tile/core.hpp"
#include "ck_tile/host/util/kernel_launch.hpp"
#include "ck_tile/host/util/kernel_launch.hpp"
#include "ck_tile/
device
/ops/rmsnorm2d.hpp"
#include "ck_tile/
host
/ops/rmsnorm2d.hpp"
#include <string>
#include <string>
template
<
typename
DataType
>
template
<
typename
DataType
>
...
...
example/ck_tile/11_add_rmsnorm2d_rdquant/add_rmsnorm2d_rdquant_fwd.hpp
View file @
f8107760
...
@@ -5,7 +5,7 @@
...
@@ -5,7 +5,7 @@
#include "ck_tile/core.hpp"
#include "ck_tile/core.hpp"
#include "ck_tile/host/util/kernel_launch.hpp"
#include "ck_tile/host/util/kernel_launch.hpp"
#include "ck_tile/
device
/ops/add_rmsnorm2d_rdquant.hpp"
#include "ck_tile/
host
/ops/add_rmsnorm2d_rdquant.hpp"
#include <string>
#include <string>
template
<
typename
DataType
>
template
<
typename
DataType
>
...
...
example/ck_tile/11_add_rmsnorm2d_rdquant/example_add_rmsnorm2d_rdquant_fwd.cpp
View file @
f8107760
#include "ck_tile/host.hpp"
#include "ck_tile/host.hpp"
#include "ck_tile/core.hpp"
#include "ck_tile/core.hpp"
#include "ck_tile/host/util/kernel_launch.hpp"
#include "ck_tile/host/util/kernel_launch.hpp"
#include "ck_tile/
device
/ops/add_rmsnorm2d_rdquant.hpp"
#include "ck_tile/
host
/ops/add_rmsnorm2d_rdquant.hpp"
#include <cstring>
#include <cstring>
// different threshold for different dtype
// different threshold for different dtype
...
...
example/ck_tile/12_smoothquant/example_smoothquant.cpp
View file @
f8107760
#include "ck_tile/host.hpp"
#include "ck_tile/host.hpp"
#include "ck_tile/core.hpp"
#include "ck_tile/core.hpp"
#include "ck_tile/host/util/kernel_launch.hpp"
#include "ck_tile/host/util/kernel_launch.hpp"
#include "ck_tile/
device
/ops/smoothquant.hpp"
#include "ck_tile/
host
/ops/smoothquant.hpp"
#include <cstring>
#include <cstring>
// different threshold for different dtype
// different threshold for different dtype
...
...
example/ck_tile/12_smoothquant/smoothquant.hpp
View file @
f8107760
...
@@ -5,7 +5,7 @@
...
@@ -5,7 +5,7 @@
#include "ck_tile/core.hpp"
#include "ck_tile/core.hpp"
#include "ck_tile/host/util/kernel_launch.hpp"
#include "ck_tile/host/util/kernel_launch.hpp"
#include "ck_tile/
device
/ops/smoothquant.hpp"
#include "ck_tile/
host
/ops/smoothquant.hpp"
#include <string>
#include <string>
template
<
typename
DataType
>
template
<
typename
DataType
>
...
...
example/ck_tile/13_moe_sorting/moe_sorting_api.hpp
View file @
f8107760
...
@@ -5,7 +5,7 @@
...
@@ -5,7 +5,7 @@
#include <string>
#include <string>
#include "ck_tile/core.hpp"
#include "ck_tile/core.hpp"
#include "ck_tile/host.hpp"
#include "ck_tile/host.hpp"
#include "ck_tile/
device
/ops/fused_moe.hpp"
#include "ck_tile/
host
/ops/fused_moe.hpp"
struct
moe_sorting_trait
struct
moe_sorting_trait
{
{
...
...
example/ck_tile/14_moe_smoothquant/moe_smoothquant.hpp
View file @
f8107760
...
@@ -5,7 +5,7 @@
...
@@ -5,7 +5,7 @@
#include "ck_tile/core.hpp"
#include "ck_tile/core.hpp"
#include "ck_tile/host/util/kernel_launch.hpp"
#include "ck_tile/host/util/kernel_launch.hpp"
#include "ck_tile/
device
/ops/smoothquant.hpp"
#include "ck_tile/
host
/ops/smoothquant.hpp"
#include <string>
#include <string>
template
<
typename
DataType
>
template
<
typename
DataType
>
...
...
example/ck_tile/15_fused_moe/fused_moegemm.hpp
View file @
f8107760
...
@@ -5,7 +5,7 @@
...
@@ -5,7 +5,7 @@
#include "ck_tile/core.hpp"
#include "ck_tile/core.hpp"
#include "ck_tile/host/util/kernel_launch.hpp"
#include "ck_tile/host/util/kernel_launch.hpp"
#include "ck_tile/
device
/ops/fused_moe.hpp"
#include "ck_tile/
host
/ops/fused_moe.hpp"
#include <string>
#include <string>
// this is only a convenient structure for creating an example
// this is only a convenient structure for creating an example
...
...
example/ck_tile/15_fused_moe/fused_moesorting.hpp
View file @
f8107760
...
@@ -5,7 +5,7 @@
...
@@ -5,7 +5,7 @@
#include <string>
#include <string>
#include "ck_tile/core.hpp"
#include "ck_tile/core.hpp"
#include "ck_tile/host.hpp"
#include "ck_tile/host.hpp"
#include "ck_tile/
device
/ops/fused_moe.hpp"
#include "ck_tile/
host
/ops/fused_moe.hpp"
struct
fused_moesorting_trait
struct
fused_moesorting_trait
{
{
...
...
example/ck_tile/15_fused_moe/instances/fused_moegemm_api_internal.hpp
View file @
f8107760
...
@@ -4,7 +4,7 @@
...
@@ -4,7 +4,7 @@
#pragma once
#pragma once
#include "fused_moegemm_api_traits.hpp"
#include "fused_moegemm_api_traits.hpp"
#include "ck_tile/
device
/ops/fused_moe.hpp"
#include "ck_tile/
host
/ops/fused_moe.hpp"
#include <iostream>
#include <iostream>
template
<
ck_tile
::
index_t
...
Is
>
template
<
ck_tile
::
index_t
...
Is
>
...
...
include/ck_tile/device/ops/layernorm2d/pipeline/layernorm2d_fwd_pipeline_default_policy.hpp
View file @
f8107760
...
@@ -4,8 +4,8 @@
...
@@ -4,8 +4,8 @@
#pragma once
#pragma once
#include "ck_tile/core.hpp"
#include "ck_tile/core.hpp"
#include "ck_tile/
device
/ops/norm_reduce/block/block_norm_reduce_problem.hpp"
#include "ck_tile/
host
/ops/norm_reduce/block/block_norm_reduce_problem.hpp"
#include "ck_tile/
device
/ops/norm_reduce/block/block_norm_reduce.hpp"
#include "ck_tile/
host
/ops/norm_reduce/block/block_norm_reduce.hpp"
namespace
ck_tile
{
namespace
ck_tile
{
...
...
include/ck_tile/host/ops/add_rmsnorm2d_rdquant/pipeline/add_rmsnorm2d_rdquant_fwd_pipeline_one_pass.hpp
View file @
f8107760
...
@@ -4,7 +4,7 @@
...
@@ -4,7 +4,7 @@
#pragma once
#pragma once
#include "ck_tile/core.hpp"
#include "ck_tile/core.hpp"
#include "ck_tile/
device
/ops/rmsnorm2d/pipeline/rmsnorm2d_fwd_pipeline_default_policy.hpp"
#include "ck_tile/
host
/ops/rmsnorm2d/pipeline/rmsnorm2d_fwd_pipeline_default_policy.hpp"
#include <string>
#include <string>
#include <type_traits>
#include <type_traits>
...
...
include/ck_tile/host/ops/add_rmsnorm2d_rdquant/pipeline/add_rmsnorm2d_rdquant_fwd_pipeline_three_pass.hpp
View file @
f8107760
...
@@ -4,7 +4,7 @@
...
@@ -4,7 +4,7 @@
#pragma once
#pragma once
#include "ck_tile/core.hpp"
#include "ck_tile/core.hpp"
#include "ck_tile/
device
/ops/rmsnorm2d/pipeline/rmsnorm2d_fwd_pipeline_default_policy.hpp"
#include "ck_tile/
host
/ops/rmsnorm2d/pipeline/rmsnorm2d_fwd_pipeline_default_policy.hpp"
#include <string>
#include <string>
#include <type_traits>
#include <type_traits>
...
...
include/ck_tile/host/ops/flatmm/block/flatmm_32x512x128_1x4x1_16x16x32.hpp
View file @
f8107760
...
@@ -5,7 +5,7 @@
...
@@ -5,7 +5,7 @@
#include "ck_tile/core.hpp"
#include "ck_tile/core.hpp"
#include "ck_tile/device/ops/gemm/warp/warp_gemm.hpp"
#include "ck_tile/device/ops/gemm/warp/warp_gemm.hpp"
#include "ck_tile/
device
/ops/flatmm/block/flatmm_uk_config.hpp"
#include "ck_tile/
host
/ops/flatmm/block/flatmm_uk_config.hpp"
namespace
ck_tile
{
namespace
ck_tile
{
...
...
include/ck_tile/host/ops/flatmm/block/flatmm_sn_32x128x512_1x4x1_16x16x32.hpp
View file @
f8107760
...
@@ -5,7 +5,7 @@
...
@@ -5,7 +5,7 @@
#include "ck_tile/core.hpp"
#include "ck_tile/core.hpp"
#include "ck_tile/device/ops/gemm/warp/warp_gemm.hpp"
#include "ck_tile/device/ops/gemm/warp/warp_gemm.hpp"
#include "ck_tile/
device
/ops/flatmm/block/flatmm_uk_config.hpp"
#include "ck_tile/
host
/ops/flatmm/block/flatmm_uk_config.hpp"
namespace
ck_tile
{
namespace
ck_tile
{
...
...
include/ck_tile/host/ops/flatmm/block/flatmm_sn_32x128x512_1x4x1_16x16x32_itl.hpp
View file @
f8107760
...
@@ -5,8 +5,8 @@
...
@@ -5,8 +5,8 @@
#include "ck_tile/core.hpp"
#include "ck_tile/core.hpp"
#include "ck_tile/device/ops/gemm/warp/warp_gemm.hpp"
#include "ck_tile/device/ops/gemm/warp/warp_gemm.hpp"
#include "ck_tile/
device
/ops/flatmm/block/flatmm_uk_config.hpp"
#include "ck_tile/
host
/ops/flatmm/block/flatmm_uk_config.hpp"
#include "ck_tile/
device
/ops/flatmm/block/flatmm_sn_32x128x512_1x4x1_16x16x32.hpp"
#include "ck_tile/
host
/ops/flatmm/block/flatmm_sn_32x128x512_1x4x1_16x16x32.hpp"
namespace
ck_tile
{
namespace
ck_tile
{
...
...
Prev
1
2
Next
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment