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
ec1b65e7
Commit
ec1b65e7
authored
Jan 19, 2025
by
Aleksander Dudek
Browse files
[CK_TILE] Update the file structure
parent
aa30ef56
Changes
262
Hide whitespace changes
Inline
Side-by-side
Showing
20 changed files
with
61 additions
and
29 deletions
+61
-29
include/ck_tile/host/ops/add_rmsnorm2d_rdquant/pipeline/add_rmsnorm2d_rdquant_fwd_pipeline_default_policy.hpp
...ine/add_rmsnorm2d_rdquant_fwd_pipeline_default_policy.hpp
+3
-3
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
+2
-2
include/ck_tile/host/ops/add_rmsnorm2d_rdquant/pipeline/add_rmsnorm2d_rdquant_fwd_pipeline_problem.hpp
...t/pipeline/add_rmsnorm2d_rdquant_fwd_pipeline_problem.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
+2
-2
include/ck_tile/host/ops/elementwise.hpp
include/ck_tile/host/ops/elementwise.hpp
+6
-0
include/ck_tile/host/ops/elementwise/unary_element_wise_operation.hpp
...ile/host/ops/elementwise/unary_element_wise_operation.hpp
+1
-1
include/ck_tile/host/ops/flatmm.hpp
include/ck_tile/host/ops/flatmm.hpp
+9
-0
include/ck_tile/host/ops/flatmm/block/flatmm_32x512x128_1x4x1_16x16x32.hpp
...ost/ops/flatmm/block/flatmm_32x512x128_1x4x1_16x16x32.hpp
+3
-3
include/ck_tile/host/ops/flatmm/block/flatmm_sn_32x128x512_1x4x1_16x16x32.hpp
.../ops/flatmm/block/flatmm_sn_32x128x512_1x4x1_16x16x32.hpp
+3
-3
include/ck_tile/host/ops/flatmm/block/flatmm_sn_32x128x512_1x4x1_16x16x32_itl.hpp
.../flatmm/block/flatmm_sn_32x128x512_1x4x1_16x16x32_itl.hpp
+4
-4
include/ck_tile/host/ops/flatmm/block/flatmm_uk_config.hpp
include/ck_tile/host/ops/flatmm/block/flatmm_uk_config.hpp
+1
-1
include/ck_tile/host/ops/flatmm/block/uk/README.md
include/ck_tile/host/ops/flatmm/block/uk/README.md
+0
-0
include/ck_tile/host/ops/flatmm/block/uk/flatmm_sn_uk_gfx9_32x128x512_1x4x1_16x16x16.inc
.../block/uk/flatmm_sn_uk_gfx9_32x128x512_1x4x1_16x16x16.inc
+0
-0
include/ck_tile/host/ops/flatmm/block/uk/flatmm_sn_uk_gfx9_32x128x512_1x4x1_16x16x16_itl.inc
...ck/uk/flatmm_sn_uk_gfx9_32x128x512_1x4x1_16x16x16_itl.inc
+0
-0
include/ck_tile/host/ops/flatmm/block/uk/flatmm_uk_gfx9_32x512x128_1x1x1_16x16x16.inc
...tmm/block/uk/flatmm_uk_gfx9_32x512x128_1x1x1_16x16x16.inc
+0
-0
include/ck_tile/host/ops/fused_moe.hpp
include/ck_tile/host/ops/fused_moe.hpp
+17
-0
include/ck_tile/host/ops/fused_moe/kernel/fused_moegemm_kernel.hpp
...k_tile/host/ops/fused_moe/kernel/fused_moegemm_kernel.hpp
+3
-3
include/ck_tile/host/ops/fused_moe/kernel/fused_moegemm_shape.hpp
...ck_tile/host/ops/fused_moe/kernel/fused_moegemm_shape.hpp
+1
-1
include/ck_tile/host/ops/fused_moe/kernel/fused_moegemm_tile_partitioner.hpp
...t/ops/fused_moe/kernel/fused_moegemm_tile_partitioner.hpp
+1
-1
include/ck_tile/host/ops/fused_moe/kernel/moe_sorting_kernel.hpp
.../ck_tile/host/ops/fused_moe/kernel/moe_sorting_kernel.hpp
+4
-4
No files found.
include/ck_tile/ops/add_rmsnorm2d_rdquant/pipeline/add_rmsnorm2d_rdquant_fwd_pipeline_default_policy.hpp
→
include/ck_tile/
host/
ops/add_rmsnorm2d_rdquant/pipeline/add_rmsnorm2d_rdquant_fwd_pipeline_default_policy.hpp
View file @
ec1b65e7
// SPDX-License-Identifier: MIT
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-202
4
, Advanced Micro Devices, Inc. All rights reserved.
// Copyright (c) 2018-202
5
, Advanced Micro Devices, Inc. All rights reserved.
#pragma once
#pragma once
#include "ck_tile/core.hpp"
#include "ck_tile/core.hpp"
#include "ck_tile/ops/reduce/block/block_reduce2d_problem.hpp"
#include "ck_tile/
device/
ops/reduce/block/block_reduce2d_problem.hpp"
#include "ck_tile/ops/reduce/block/block_reduce2d.hpp"
#include "ck_tile/
device/
ops/reduce/block/block_reduce2d.hpp"
namespace
ck_tile
{
namespace
ck_tile
{
...
...
include/ck_tile/ops/add_rmsnorm2d_rdquant/pipeline/add_rmsnorm2d_rdquant_fwd_pipeline_one_pass.hpp
→
include/ck_tile/
host/
ops/add_rmsnorm2d_rdquant/pipeline/add_rmsnorm2d_rdquant_fwd_pipeline_one_pass.hpp
View file @
ec1b65e7
// SPDX-License-Identifier: MIT
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-202
4
, Advanced Micro Devices, Inc. All rights reserved.
// Copyright (c) 2018-202
5
, Advanced Micro Devices, Inc. All rights reserved.
#pragma once
#pragma once
#include "ck_tile/core.hpp"
#include "ck_tile/core.hpp"
#include "ck_tile/ops/rmsnorm2d/pipeline/rmsnorm2d_fwd_pipeline_default_policy.hpp"
#include "ck_tile/
device/
ops/rmsnorm2d/pipeline/rmsnorm2d_fwd_pipeline_default_policy.hpp"
#include <string>
#include <string>
#include <type_traits>
#include <type_traits>
...
...
include/ck_tile/ops/add_rmsnorm2d_rdquant/pipeline/add_rmsnorm2d_rdquant_fwd_pipeline_problem.hpp
→
include/ck_tile/
host/
ops/add_rmsnorm2d_rdquant/pipeline/add_rmsnorm2d_rdquant_fwd_pipeline_problem.hpp
View file @
ec1b65e7
// SPDX-License-Identifier: MIT
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-202
4
, Advanced Micro Devices, Inc. All rights reserved.
// Copyright (c) 2018-202
5
, Advanced Micro Devices, Inc. All rights reserved.
#pragma once
#pragma once
...
...
include/ck_tile/ops/add_rmsnorm2d_rdquant/pipeline/add_rmsnorm2d_rdquant_fwd_pipeline_three_pass.hpp
→
include/ck_tile/
host/
ops/add_rmsnorm2d_rdquant/pipeline/add_rmsnorm2d_rdquant_fwd_pipeline_three_pass.hpp
View file @
ec1b65e7
// SPDX-License-Identifier: MIT
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-202
4
, Advanced Micro Devices, Inc. All rights reserved.
// Copyright (c) 2018-202
5
, Advanced Micro Devices, Inc. All rights reserved.
#pragma once
#pragma once
#include "ck_tile/core.hpp"
#include "ck_tile/core.hpp"
#include "ck_tile/ops/rmsnorm2d/pipeline/rmsnorm2d_fwd_pipeline_default_policy.hpp"
#include "ck_tile/
device/
ops/rmsnorm2d/pipeline/rmsnorm2d_fwd_pipeline_default_policy.hpp"
#include <string>
#include <string>
#include <type_traits>
#include <type_traits>
...
...
include/ck_tile/ops/elementwise.hpp
→
include/ck_tile/
host/
ops/elementwise.hpp
View file @
ec1b65e7
...
@@ -3,6 +3,4 @@
...
@@ -3,6 +3,4 @@
#pragma once
#pragma once
#include "ck_tile/ops/elementwise/unary_element_wise_operation.hpp"
#include "ck_tile/host/ops/elementwise/unary_element_wise_operation.hpp"
#include "ck_tile/ops/common/generic_2d_block_shape.hpp"
#include "ck_tile/ops/common/tensor_layout.hpp"
include/ck_tile/ops/elementwise/unary_element_wise_operation.hpp
→
include/ck_tile/
host/
ops/elementwise/unary_element_wise_operation.hpp
View file @
ec1b65e7
// SPDX-License-Identifier: MIT
// SPDX-License-Identifier: MIT
// Copyright (c) 202
4
, Advanced Micro Devices, Inc. All rights reserved.
// Copyright (c) 202
5
, Advanced Micro Devices, Inc. All rights reserved.
#pragma once
#pragma once
...
...
include/ck_tile/host/ops/flatmm.hpp
0 → 100644
View file @
ec1b65e7
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-2025, Advanced Micro Devices, Inc. All rights reserved.
#pragma once
#include "ck_tile/host/ops/flatmm/block/flatmm_32x512x128_1x4x1_16x16x32.hpp"
#include "ck_tile/host/ops/flatmm/block/flatmm_sn_32x128x512_1x4x1_16x16x32.hpp"
#include "ck_tile/host/ops/flatmm/block/flatmm_sn_32x128x512_1x4x1_16x16x32_itl.hpp"
#include "ck_tile/host/ops/flatmm/block/flatmm_uk_config.hpp"
include/ck_tile/ops/flatmm/block/flatmm_32x512x128_1x4x1_16x16x32.hpp
→
include/ck_tile/
host/
ops/flatmm/block/flatmm_32x512x128_1x4x1_16x16x32.hpp
View file @
ec1b65e7
// SPDX-License-Identifier: MIT
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-202
4
, Advanced Micro Devices, Inc. All rights reserved.
// Copyright (c) 2018-202
5
, Advanced Micro Devices, Inc. All rights reserved.
#pragma once
#pragma once
#include "ck_tile/core.hpp"
#include "ck_tile/core.hpp"
#include "ck_tile/ops/gemm/warp/warp_gemm.hpp"
#include "ck_tile/
device/
ops/gemm/warp/warp_gemm.hpp"
#include "ck_tile/ops/flatmm/block/flatmm_uk_config.hpp"
#include "ck_tile/
device/
ops/flatmm/block/flatmm_uk_config.hpp"
namespace
ck_tile
{
namespace
ck_tile
{
...
...
include/ck_tile/ops/flatmm/block/flatmm_sn_32x128x512_1x4x1_16x16x32.hpp
→
include/ck_tile/
host/
ops/flatmm/block/flatmm_sn_32x128x512_1x4x1_16x16x32.hpp
View file @
ec1b65e7
// SPDX-License-Identifier: MIT
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-202
4
, Advanced Micro Devices, Inc. All rights reserved.
// Copyright (c) 2018-202
5
, Advanced Micro Devices, Inc. All rights reserved.
#pragma once
#pragma once
#include "ck_tile/core.hpp"
#include "ck_tile/core.hpp"
#include "ck_tile/ops/gemm/warp/warp_gemm.hpp"
#include "ck_tile/
device/
ops/gemm/warp/warp_gemm.hpp"
#include "ck_tile/ops/flatmm/block/flatmm_uk_config.hpp"
#include "ck_tile/
device/
ops/flatmm/block/flatmm_uk_config.hpp"
namespace
ck_tile
{
namespace
ck_tile
{
...
...
include/ck_tile/ops/flatmm/block/flatmm_sn_32x128x512_1x4x1_16x16x32_itl.hpp
→
include/ck_tile/
host/
ops/flatmm/block/flatmm_sn_32x128x512_1x4x1_16x16x32_itl.hpp
View file @
ec1b65e7
// SPDX-License-Identifier: MIT
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-202
4
, Advanced Micro Devices, Inc. All rights reserved.
// Copyright (c) 2018-202
5
, Advanced Micro Devices, Inc. All rights reserved.
#pragma once
#pragma once
#include "ck_tile/core.hpp"
#include "ck_tile/core.hpp"
#include "ck_tile/ops/gemm/warp/warp_gemm.hpp"
#include "ck_tile/
device/
ops/gemm/warp/warp_gemm.hpp"
#include "ck_tile/ops/flatmm/block/flatmm_uk_config.hpp"
#include "ck_tile/
device/
ops/flatmm/block/flatmm_uk_config.hpp"
#include "ck_tile/ops/flatmm/block/flatmm_sn_32x128x512_1x4x1_16x16x32.hpp"
#include "ck_tile/
device/
ops/flatmm/block/flatmm_sn_32x128x512_1x4x1_16x16x32.hpp"
namespace
ck_tile
{
namespace
ck_tile
{
...
...
include/ck_tile/ops/flatmm/block/flatmm_uk_config.hpp
→
include/ck_tile/
host/
ops/flatmm/block/flatmm_uk_config.hpp
View file @
ec1b65e7
// SPDX-License-Identifier: MIT
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-202
4
, Advanced Micro Devices, Inc. All rights reserved.
// Copyright (c) 2018-202
5
, Advanced Micro Devices, Inc. All rights reserved.
#pragma once
#pragma once
...
...
include/ck_tile/ops/flatmm/block/uk/README.md
→
include/ck_tile/
host/
ops/flatmm/block/uk/README.md
View file @
ec1b65e7
File moved
include/ck_tile/ops/flatmm/block/uk/flatmm_sn_uk_gfx9_32x128x512_1x4x1_16x16x16.inc
→
include/ck_tile/
host/
ops/flatmm/block/uk/flatmm_sn_uk_gfx9_32x128x512_1x4x1_16x16x16.inc
View file @
ec1b65e7
File moved
include/ck_tile/ops/flatmm/block/uk/flatmm_sn_uk_gfx9_32x128x512_1x4x1_16x16x16_itl.inc
→
include/ck_tile/
host/
ops/flatmm/block/uk/flatmm_sn_uk_gfx9_32x128x512_1x4x1_16x16x16_itl.inc
View file @
ec1b65e7
File moved
include/ck_tile/ops/flatmm/block/uk/flatmm_uk_gfx9_32x512x128_1x1x1_16x16x16.inc
→
include/ck_tile/
host/
ops/flatmm/block/uk/flatmm_uk_gfx9_32x512x128_1x1x1_16x16x16.inc
View file @
ec1b65e7
File moved
include/ck_tile/host/ops/fused_moe.hpp
0 → 100644
View file @
ec1b65e7
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-2025, Advanced Micro Devices, Inc. All rights reserved.
#pragma once
#include "ck_tile/host/ops/fused_moe/kernel/fused_moegemm_kernel.hpp"
#include "ck_tile/host/ops/fused_moe/kernel/fused_moegemm_shape.hpp"
#include "ck_tile/host/ops/fused_moe/kernel/fused_moegemm_tile_partitioner.hpp"
#include "ck_tile/host/ops/fused_moe/kernel/moe_sorting_kernel.hpp"
#include "ck_tile/host/ops/fused_moe/pipeline/fused_moegemm_pipeline_flatmm_ex.hpp"
#include "ck_tile/host/ops/fused_moe/pipeline/fused_moegemm_pipeline_flatmm_policy.hpp"
#include "ck_tile/host/ops/fused_moe/pipeline/fused_moegemm_pipeline_flatmm_uk.hpp"
#include "ck_tile/host/ops/fused_moe/pipeline/fused_moegemm_pipeline_problem.hpp"
#include "ck_tile/host/ops/fused_moe/pipeline/fused_moegemm_traits.hpp"
#include "ck_tile/host/ops/fused_moe/pipeline/moe_sorting_pipeline.hpp"
#include "ck_tile/host/ops/fused_moe/pipeline/moe_sorting_policy.hpp"
#include "ck_tile/host/ops/fused_moe/pipeline/moe_sorting_problem.hpp"
include/ck_tile/ops/fused_moe/kernel/fused_moegemm_kernel.hpp
→
include/ck_tile/
host/
ops/fused_moe/kernel/fused_moegemm_kernel.hpp
View file @
ec1b65e7
// SPDX-License-Identifier: MIT
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-202
4
, Advanced Micro Devices, Inc. All rights reserved.
// Copyright (c) 2018-202
5
, Advanced Micro Devices, Inc. All rights reserved.
#pragma once
#pragma once
#include "ck_tile/core.hpp"
#include "ck_tile/core.hpp"
#include "ck_tile/ops/common.hpp"
#include "ck_tile/
device/
ops/common.hpp"
#include "ck_tile/ops/elementwise.hpp"
#include "ck_tile/
device/
ops/elementwise.hpp"
#include <string>
#include <string>
#include <type_traits>
#include <type_traits>
...
...
include/ck_tile/ops/fused_moe/kernel/fused_moegemm_shape.hpp
→
include/ck_tile/
host/
ops/fused_moe/kernel/fused_moegemm_shape.hpp
View file @
ec1b65e7
// SPDX-License-Identifier: MIT
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-202
3
, Advanced Micro Devices, Inc. All rights reserved.
// Copyright (c) 2018-202
5
, Advanced Micro Devices, Inc. All rights reserved.
#pragma once
#pragma once
...
...
include/ck_tile/ops/fused_moe/kernel/fused_moegemm_tile_partitioner.hpp
→
include/ck_tile/
host/
ops/fused_moe/kernel/fused_moegemm_tile_partitioner.hpp
View file @
ec1b65e7
// SPDX-License-Identifier: MIT
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-202
4
, Advanced Micro Devices, Inc. All rights reserved.
// Copyright (c) 2018-202
5
, Advanced Micro Devices, Inc. All rights reserved.
#pragma once
#pragma once
...
...
include/ck_tile/ops/fused_moe/kernel/moe_sorting_kernel.hpp
→
include/ck_tile/
host/
ops/fused_moe/kernel/moe_sorting_kernel.hpp
View file @
ec1b65e7
// SPDX-License-Identifier: MIT
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-202
4
, Advanced Micro Devices, Inc. All rights reserved.
// Copyright (c) 2018-202
5
, Advanced Micro Devices, Inc. All rights reserved.
#pragma once
#pragma once
#include "ck_tile/core.hpp"
#include "ck_tile/core.hpp"
#include "ck_tile/ops/common.hpp"
#include "ck_tile/
device/
ops/common.hpp"
#include "ck_tile/ops/elementwise.hpp"
#include "ck_tile/
device/
ops/elementwise.hpp"
#include "ck_tile/host/hip_check_error.hpp"
#include "ck_tile/host/
util/
hip_check_error.hpp"
#include <string>
#include <string>
#include <type_traits>
#include <type_traits>
...
...
Prev
1
…
4
5
6
7
8
9
10
11
12
…
14
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