Skip to content

GitLab

Commit ec1b65e7 authored by Aleksander Dudek's avatar Aleksander Dudek
Browse files

[CK_TILE] Update the file structure

parent aa30ef56
Hide whitespace changes
Inline Side-by-side
Showing with 64 additions and 37 deletions
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 // SPDX-License-Identifier: MIT
// Copyright (c) 2018-2024, Advanced Micro Devices, Inc. All rights reserved. // Copyright (c) 2018-2025, 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/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 <string>
#include <type_traits> #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 // SPDX-License-Identifier: MIT
// Copyright (c) 2018-2024, Advanced Micro Devices, Inc. All rights reserved. // Copyright (c) 2018-2025, 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/softmax.hpp" #include "ck_tile/device/ops/softmax.hpp"
#include "ck_tile/ops/topk.hpp" #include "ck_tile/device/ops/topk.hpp"
namespace ck_tile { 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 // SPDX-License-Identifier: MIT
// Copyright (c) 2018-2024, Advanced Micro Devices, Inc. All rights reserved. // Copyright (c) 2018-2025, Advanced Micro Devices, Inc. All rights reserved.
#pragma once #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/naive_attention.hpp → include/ck_tile/host/reference/naive_attention.hpp
View file @ ec1b65e7
// SPDX-License-Identifier: MIT // SPDX-License-Identifier: MIT
// Copyright (c) 2018-2024, Advanced Micro Devices, Inc. All rights reserved. // Copyright (c) 2018-2025, Advanced Micro Devices, Inc. All rights reserved.
#pragma once #pragma once
#include "ck_tile/core.hpp" #include "ck_tile/core.hpp"
#include "ck_tile/host/host_tensor.hpp" #include "ck_tile/host/util/host_tensor.hpp"
#include "ck_tile/host/kernel_launch.hpp" #include "ck_tile/host/util/kernel_launch.hpp"
#include <thread> #include <thread>
#include <string> #include <string>
......
include/ck_tile/host/reference/reference_batched_dropout.hpp
View file @ ec1b65e7
// SPDX-License-Identifier: MIT // SPDX-License-Identifier: MIT
// Copyright (c) 2018-2024, Advanced Micro Devices, Inc. All rights reserved. // Copyright (c) 2018-2025, Advanced Micro Devices, Inc. All rights reserved.
#pragma once #pragma once
#include "ck_tile/core.hpp" #include "ck_tile/core.hpp"
#include "ck_tile/host/host_tensor.hpp" #include "ck_tile/host/util/host_tensor.hpp"
#include <thread> #include <thread>
namespace ck_tile { namespace ck_tile {
......
include/ck_tile/host/reference/reference_batched_elementwise.hpp
View file @ ec1b65e7
// SPDX-License-Identifier: MIT // SPDX-License-Identifier: MIT
// Copyright (c) 2018-2023, Advanced Micro Devices, Inc. All rights reserved. // Copyright (c) 2018-2025, Advanced Micro Devices, Inc. All rights reserved.
#pragma once #pragma once
#include "ck_tile/core.hpp" #include "ck_tile/core.hpp"
#include "ck_tile/host/host_tensor.hpp" #include "ck_tile/host/util/host_tensor.hpp"
#include <thread> #include <thread>
namespace ck_tile { namespace ck_tile {
......
include/ck_tile/host/reference/reference_batched_gemm.hpp
View file @ ec1b65e7
// SPDX-License-Identifier: MIT // SPDX-License-Identifier: MIT
// Copyright (c) 2018-2023, Advanced Micro Devices, Inc. All rights reserved. // Copyright (c) 2018-2025, Advanced Micro Devices, Inc. All rights reserved.
#pragma once #pragma once
#include "ck_tile/core.hpp" #include "ck_tile/core.hpp"
#include "ck_tile/host/host_tensor.hpp" #include "ck_tile/host/util/host_tensor.hpp"
#include <thread> #include <thread>
namespace ck_tile { namespace ck_tile {
......
include/ck_tile/host/reference/reference_batched_masking.hpp
View file @ ec1b65e7
// SPDX-License-Identifier: MIT // SPDX-License-Identifier: MIT
// Copyright (c) 2018-2023, Advanced Micro Devices, Inc. All rights reserved. // Copyright (c) 2018-2025, Advanced Micro Devices, Inc. All rights reserved.
#pragma once #pragma once
#include "ck_tile/core.hpp" #include "ck_tile/core.hpp"
#include "ck_tile/host/host_tensor.hpp" #include "ck_tile/host/util/host_tensor.hpp"
#include <thread> #include <thread>
namespace ck_tile { namespace ck_tile {
......
include/ck_tile/host/reference/reference_batched_rotary_position_embedding.hpp
View file @ ec1b65e7
// SPDX-License-Identifier: MIT // SPDX-License-Identifier: MIT
// Copyright (c) 2018-2024, Advanced Micro Devices, Inc. All rights reserved. // Copyright (c) 2018-2025, Advanced Micro Devices, Inc. All rights reserved.
#pragma once #pragma once
#include "ck_tile/core.hpp" #include "ck_tile/core.hpp"
#include "ck_tile/host/host_tensor.hpp" #include "ck_tile/host/util/host_tensor.hpp"
#include <cassert> #include <cassert>
#include <thread> #include <thread>
......
include/ck_tile/host/reference/reference_batched_softmax.hpp
View file @ ec1b65e7
// SPDX-License-Identifier: MIT // SPDX-License-Identifier: MIT
// Copyright (c) 2018-2023, Advanced Micro Devices, Inc. All rights reserved. // Copyright (c) 2018-2025, Advanced Micro Devices, Inc. All rights reserved.
#pragma once #pragma once
#include "ck_tile/core.hpp" #include "ck_tile/core.hpp"
#include "ck_tile/host/host_tensor.hpp" #include "ck_tile/host/util/host_tensor.hpp"
#include <thread> #include <thread>
#include <optional>
namespace ck_tile { namespace ck_tile {
......
include/ck_tile/host/reference/reference_elementwise.hpp
View file @ ec1b65e7
// SPDX-License-Identifier: MIT // SPDX-License-Identifier: MIT
// Copyright (c) 2018-2024, Advanced Micro Devices, Inc. All rights reserved. // Copyright (c) 2018-2025, Advanced Micro Devices, Inc. All rights reserved.
#pragma once #pragma once
#include "ck_tile/core.hpp" #include "ck_tile/core.hpp"
#include "ck_tile/host/host_tensor.hpp" #include "ck_tile/host/util/host_tensor.hpp"
#include <thread> #include <thread>
namespace ck_tile { namespace ck_tile {
......
include/ck_tile/host/reference/reference_fused_moe.hpp
View file @ ec1b65e7
// SPDX-License-Identifier: MIT // SPDX-License-Identifier: MIT
// Copyright (c) 2024, Advanced Micro Devices, Inc. All rights reserved. // Copyright (c) 2025, Advanced Micro Devices, Inc. All rights reserved.
#pragma once #pragma once
#include "ck_tile/core.hpp" #include "ck_tile/core.hpp"
#include "ck_tile/host/host_tensor.hpp" #include "ck_tile/host/util/host_tensor.hpp"
namespace ck_tile { namespace ck_tile {
// [indexing implementation-1] // [indexing implementation-1]
......
include/ck_tile/host/reference/reference_gemm.hpp
View file @ ec1b65e7
// SPDX-License-Identifier: MIT // SPDX-License-Identifier: MIT
// Copyright (c) 2018-2024, Advanced Micro Devices, Inc. All rights reserved. // Copyright (c) 2018-2025, Advanced Micro Devices, Inc. All rights reserved.
#pragma once #pragma once
...@@ -7,7 +7,8 @@ ...@@ -7,7 +7,8 @@
#include <thread> #include <thread>
#include "ck_tile/core.hpp" #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 { namespace ck_tile {
......
include/ck_tile/host/reference/reference_im2col.hpp
View file @ ec1b65e7
// SPDX-License-Identifier: MIT // SPDX-License-Identifier: MIT
// Copyright (c) 2018-2024, Advanced Micro Devices, Inc. All rights reserved. // Copyright (c) 2018-2025, Advanced Micro Devices, Inc. All rights reserved.
#pragma once #pragma once
#include "ck_tile/core.hpp" #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> #include <thread>
namespace ck_tile { namespace ck_tile {
......
include/ck_tile/host/reference/reference_layernorm2d_fwd.hpp
View file @ ec1b65e7
// SPDX-License-Identifier: MIT // SPDX-License-Identifier: MIT
// Copyright (c) 2018-2023, Advanced Micro Devices, Inc. All rights reserved. // Copyright (c) 2018-2025, Advanced Micro Devices, Inc. All rights reserved.
#pragma once #pragma once
#include "ck_tile/core.hpp" #include "ck_tile/core.hpp"
#include "ck_tile/host/host_tensor.hpp" #include "ck_tile/host/util/host_tensor.hpp"
namespace ck_tile { namespace ck_tile {
......
include/ck_tile/host/reference/reference_moe_sorting.hpp
View file @ ec1b65e7
// SPDX-License-Identifier: MIT // SPDX-License-Identifier: MIT
// Copyright (c) 2024, Advanced Micro Devices, Inc. All rights reserved. // Copyright (c) 2025, Advanced Micro Devices, Inc. All rights reserved.
#pragma once #pragma once
#include "ck_tile/core.hpp" #include "ck_tile/core.hpp"
#include "ck_tile/host/host_tensor.hpp" #include "ck_tile/host/util/host_tensor.hpp"
namespace ck_tile { namespace ck_tile {
......
include/ck_tile/host/reference/reference_permute.hpp
View file @ ec1b65e7
// SPDX-License-Identifier: MIT // SPDX-License-Identifier: MIT
// Copyright (c) 2018-2024, Advanced Micro Devices, Inc. All rights reserved. // Copyright (c) 2018-2025, Advanced Micro Devices, Inc. All rights reserved.
#pragma once #pragma once
#include "ck_tile/core.hpp" #include "ck_tile/core.hpp"
#include "ck_tile/host/host_tensor.hpp" #include "ck_tile/host/util/host_tensor.hpp"
#include <thread> #include <thread>
#include <numeric> #include <numeric>
#include <functional> #include <functional>
......
include/ck_tile/host/reference/reference_reduce.hpp
View file @ ec1b65e7
// SPDX-License-Identifier: MIT // SPDX-License-Identifier: MIT
// Copyright (c) 2018-2023, Advanced Micro Devices, Inc. All rights reserved. // Copyright (c) 2018-2025, Advanced Micro Devices, Inc. All rights reserved.
#pragma once #pragma once
#include "ck_tile/core.hpp" #include "ck_tile/core.hpp"
#include "ck_tile/host/host_tensor.hpp" #include "ck_tile/host/util/host_tensor.hpp"
#include <thread> #include <thread>
namespace ck_tile { namespace ck_tile {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment