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
64 additions
and
37 deletions
+64
-37
include/ck_tile/host/ops/topk_softmax/pipeline/topk_softmax_warp_per_row_pipeline.hpp
...k_softmax/pipeline/topk_softmax_warp_per_row_pipeline.hpp
+2
-2
include/ck_tile/host/ops/topk_softmax/pipeline/topk_softmax_warp_per_row_policy.hpp
...opk_softmax/pipeline/topk_softmax_warp_per_row_policy.hpp
+3
-3
include/ck_tile/host/ops/topk_softmax/pipeline/topk_softmax_warp_per_row_problem.hpp
...pk_softmax/pipeline/topk_softmax_warp_per_row_problem.hpp
+1
-1
include/ck_tile/host/reference.hpp
include/ck_tile/host/reference.hpp
+24
-0
include/ck_tile/host/reference/README.md
include/ck_tile/host/reference/README.md
+0
-0
include/ck_tile/host/reference/naive_attention.hpp
include/ck_tile/host/reference/naive_attention.hpp
+3
-3
include/ck_tile/host/reference/reference_batched_dropout.hpp
include/ck_tile/host/reference/reference_batched_dropout.hpp
+2
-2
include/ck_tile/host/reference/reference_batched_elementwise.hpp
.../ck_tile/host/reference/reference_batched_elementwise.hpp
+2
-2
include/ck_tile/host/reference/reference_batched_gemm.hpp
include/ck_tile/host/reference/reference_batched_gemm.hpp
+2
-2
include/ck_tile/host/reference/reference_batched_masking.hpp
include/ck_tile/host/reference/reference_batched_masking.hpp
+2
-2
include/ck_tile/host/reference/reference_batched_rotary_position_embedding.hpp
...reference/reference_batched_rotary_position_embedding.hpp
+2
-2
include/ck_tile/host/reference/reference_batched_softmax.hpp
include/ck_tile/host/reference/reference_batched_softmax.hpp
+3
-2
include/ck_tile/host/reference/reference_elementwise.hpp
include/ck_tile/host/reference/reference_elementwise.hpp
+2
-2
include/ck_tile/host/reference/reference_fused_moe.hpp
include/ck_tile/host/reference/reference_fused_moe.hpp
+2
-2
include/ck_tile/host/reference/reference_gemm.hpp
include/ck_tile/host/reference/reference_gemm.hpp
+3
-2
include/ck_tile/host/reference/reference_im2col.hpp
include/ck_tile/host/reference/reference_im2col.hpp
+3
-2
include/ck_tile/host/reference/reference_layernorm2d_fwd.hpp
include/ck_tile/host/reference/reference_layernorm2d_fwd.hpp
+2
-2
include/ck_tile/host/reference/reference_moe_sorting.hpp
include/ck_tile/host/reference/reference_moe_sorting.hpp
+2
-2
include/ck_tile/host/reference/reference_permute.hpp
include/ck_tile/host/reference/reference_permute.hpp
+2
-2
include/ck_tile/host/reference/reference_reduce.hpp
include/ck_tile/host/reference/reference_reduce.hpp
+2
-2
No files found.
include/ck_tile/ops/topk_softmax/pipeline/topk_softmax_warp_per_row_pipeline.hpp
→
include/ck_tile/
host/
ops/topk_softmax/pipeline/topk_softmax_warp_per_row_pipeline.hpp
View file @
ec1b65e7
// 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
#include "ck_tile/core.hpp"
#include "ck_tile/ops/topk_softmax/pipeline/topk_softmax_warp_per_row_policy.hpp"
#include "ck_tile/
device/
ops/topk_softmax/pipeline/topk_softmax_warp_per_row_policy.hpp"
#include <string>
#include <type_traits>
...
...
include/ck_tile/ops/topk_softmax/pipeline/topk_softmax_warp_per_row_policy.hpp
→
include/ck_tile/
host/
ops/topk_softmax/pipeline/topk_softmax_warp_per_row_policy.hpp
View file @
ec1b65e7
// 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
#include "ck_tile/core.hpp"
#include "ck_tile/ops/softmax.hpp"
#include "ck_tile/ops/topk.hpp"
#include "ck_tile/
device/
ops/softmax.hpp"
#include "ck_tile/
device/
ops/topk.hpp"
namespace
ck_tile
{
...
...
include/ck_tile/ops/topk_softmax/pipeline/topk_softmax_warp_per_row_problem.hpp
→
include/ck_tile/
host/
ops/topk_softmax/pipeline/topk_softmax_warp_per_row_problem.hpp
View file @
ec1b65e7
// 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
...
...
include/ck_tile/host/reference.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/reference/naive_attention.hpp"
#include "ck_tile/host/reference/reference_batched_dropout.hpp"
#include "ck_tile/host/reference/reference_batched_elementwise.hpp"
#include "ck_tile/host/reference/reference_batched_gemm.hpp"
#include "ck_tile/host/reference/reference_batched_masking.hpp"
#include "ck_tile/host/reference/reference_batched_rotary_position_embedding.hpp"
#include "ck_tile/host/reference/reference_batched_softmax.hpp"
#include "ck_tile/host/reference/reference_elementwise.hpp"
#include "ck_tile/host/reference/reference_fused_moe.hpp"
#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_moe_sorting.hpp"
#include "ck_tile/host/reference/reference_permute.hpp"
#include "ck_tile/host/reference/reference_reduce.hpp"
#include "ck_tile/host/reference/reference_rmsnorm2d_fwd.hpp"
#include "ck_tile/host/reference/reference_rowwise_quantization2d.hpp"
#include "ck_tile/host/reference/reference_softmax.hpp"
#include "ck_tile/host/reference/reference_topk.hpp"
include/ck_tile/
ref
/README.md
→
include/ck_tile/
host/reference
/README.md
View file @
ec1b65e7
File moved
include/ck_tile/
ref
/naive_attention.hpp
→
include/ck_tile/
host/reference
/naive_attention.hpp
View file @
ec1b65e7
// 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
#include "ck_tile/core.hpp"
#include "ck_tile/host/host_tensor.hpp"
#include "ck_tile/host/kernel_launch.hpp"
#include "ck_tile/host/
util/
host_tensor.hpp"
#include "ck_tile/host/
util/
kernel_launch.hpp"
#include <thread>
#include <string>
...
...
include/ck_tile/host/reference/reference_batched_dropout.hpp
View file @
ec1b65e7
// 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
#include "ck_tile/core.hpp"
#include "ck_tile/host/host_tensor.hpp"
#include "ck_tile/host/
util/
host_tensor.hpp"
#include <thread>
namespace
ck_tile
{
...
...
include/ck_tile/host/reference/reference_batched_elementwise.hpp
View file @
ec1b65e7
// 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
#include "ck_tile/core.hpp"
#include "ck_tile/host/host_tensor.hpp"
#include "ck_tile/host/
util/
host_tensor.hpp"
#include <thread>
namespace
ck_tile
{
...
...
include/ck_tile/host/reference/reference_batched_gemm.hpp
View file @
ec1b65e7
// 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
#include "ck_tile/core.hpp"
#include "ck_tile/host/host_tensor.hpp"
#include "ck_tile/host/
util/
host_tensor.hpp"
#include <thread>
namespace
ck_tile
{
...
...
include/ck_tile/host/reference/reference_batched_masking.hpp
View file @
ec1b65e7
// 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
#include "ck_tile/core.hpp"
#include "ck_tile/host/host_tensor.hpp"
#include "ck_tile/host/
util/
host_tensor.hpp"
#include <thread>
namespace
ck_tile
{
...
...
include/ck_tile/host/reference/reference_batched_rotary_position_embedding.hpp
View file @
ec1b65e7
// 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
#include "ck_tile/core.hpp"
#include "ck_tile/host/host_tensor.hpp"
#include "ck_tile/host/
util/
host_tensor.hpp"
#include <cassert>
#include <thread>
...
...
include/ck_tile/host/reference/reference_batched_softmax.hpp
View file @
ec1b65e7
// 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
#include "ck_tile/core.hpp"
#include "ck_tile/host/host_tensor.hpp"
#include "ck_tile/host/
util/
host_tensor.hpp"
#include <thread>
#include <optional>
namespace
ck_tile
{
...
...
include/ck_tile/host/reference/reference_elementwise.hpp
View file @
ec1b65e7
// 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
#include "ck_tile/core.hpp"
#include "ck_tile/host/host_tensor.hpp"
#include "ck_tile/host/
util/
host_tensor.hpp"
#include <thread>
namespace
ck_tile
{
...
...
include/ck_tile/host/reference/reference_fused_moe.hpp
View file @
ec1b65e7
// 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
#include "ck_tile/core.hpp"
#include "ck_tile/host/host_tensor.hpp"
#include "ck_tile/host/
util/
host_tensor.hpp"
namespace
ck_tile
{
// [indexing implementation-1]
...
...
include/ck_tile/host/reference/reference_gemm.hpp
View file @
ec1b65e7
// 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
...
...
@@ -7,7 +7,8 @@
#include <thread>
#include "ck_tile/core.hpp"
#include "ck_tile/host/host_tensor.hpp"
#include "ck_tile/host/util/host_tensor.hpp"
#include "ck_tile/device/ops/common/tensor_layout.hpp"
namespace
ck_tile
{
...
...
include/ck_tile/host/reference/reference_im2col.hpp
View file @
ec1b65e7
// 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
#include "ck_tile/core.hpp"
#include "ck_tile/host/host_tensor.hpp"
#include "ck_tile/host/util/host_tensor.hpp"
#include "ck_tile/host/util/convolution_parameter.hpp"
#include <thread>
namespace
ck_tile
{
...
...
include/ck_tile/host/reference/reference_layernorm2d_fwd.hpp
View file @
ec1b65e7
// 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
#include "ck_tile/core.hpp"
#include "ck_tile/host/host_tensor.hpp"
#include "ck_tile/host/
util/
host_tensor.hpp"
namespace
ck_tile
{
...
...
include/ck_tile/host/reference/reference_moe_sorting.hpp
View file @
ec1b65e7
// 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
#include "ck_tile/core.hpp"
#include "ck_tile/host/host_tensor.hpp"
#include "ck_tile/host/
util/
host_tensor.hpp"
namespace
ck_tile
{
...
...
include/ck_tile/host/reference/reference_permute.hpp
View file @
ec1b65e7
// 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
#include "ck_tile/core.hpp"
#include "ck_tile/host/host_tensor.hpp"
#include "ck_tile/host/
util/
host_tensor.hpp"
#include <thread>
#include <numeric>
#include <functional>
...
...
include/ck_tile/host/reference/reference_reduce.hpp
View file @
ec1b65e7
// 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
#include "ck_tile/core.hpp"
#include "ck_tile/host/host_tensor.hpp"
#include "ck_tile/host/
util/
host_tensor.hpp"
#include <thread>
namespace
ck_tile
{
...
...
Prev
1
…
7
8
9
10
11
12
13
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