"git@developer.sourcefind.cn:gaoqiong/composable_kernel.git" did not exist on "7b30d202e29d18ea3edb4e1066e86f69e0a4ca33"
Commit ec1b65e7 authored by Aleksander Dudek's avatar Aleksander Dudek
Browse files

[CK_TILE] Update the file structure

parent aa30ef56
// 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 <ostream> #include <ostream>
#include <string> #include <string>
#include "ck_tile/core.hpp" #include "ck_tile/core.hpp"
#include "ck_tile/ops/fmha.hpp" #include "ck_tile/device/ops/fmha.hpp"
// keep sync with BlockAttentionBiasEnum // keep sync with BlockAttentionBiasEnum
enum class bias_enum enum class bias_enum
......
// 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/kernel_launch.hpp" #include "ck_tile/host/util/kernel_launch.hpp"
#include "ck_tile/ops/fmha.hpp" #include "ck_tile/device/ops/fmha.hpp"
#include "ck_tile/ops/epilogue.hpp" #include "ck_tile/device/ops/epilogue.hpp"
#include "mask.hpp" #include "mask.hpp"
#include "bias.hpp" #include "bias.hpp"
......
// 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/kernel_launch.hpp" #include "ck_tile/host/util/kernel_launch.hpp"
#include "ck_tile/ops/epilogue.hpp" #include "ck_tile/device/ops/epilogue.hpp"
#include "ck_tile/ops/fmha.hpp" #include "ck_tile/device/ops/fmha.hpp"
#include "bias.hpp" #include "bias.hpp"
#include "mask.hpp" #include "mask.hpp"
......
// 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,7 @@ ...@@ -7,7 +7,7 @@
#include <string> #include <string>
#include "ck_tile/core.hpp" #include "ck_tile/core.hpp"
#include "ck_tile/ops/fmha.hpp" #include "ck_tile/device/ops/fmha.hpp"
// keep this in sync with ck_tile::GenericAttentionMaskEnum // keep this in sync with ck_tile::GenericAttentionMaskEnum
enum class mask_enum enum class mask_enum
......
// 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 <cmath> #include <cmath>
......
// 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/kernel_launch.hpp" #include "ck_tile/host/util/kernel_launch.hpp"
#include "ck_tile/ops/layernorm2d.hpp" #include "ck_tile/device/ops/layernorm2d.hpp"
#include <string> #include <string>
template <typename InType, typename OutType, typename XScaleDataType_, typename YScaleDataType_> template <typename InType, typename OutType, typename XScaleDataType_, typename YScaleDataType_>
......
...@@ -6,9 +6,9 @@ ...@@ -6,9 +6,9 @@
#include <string> #include <string>
#include "ck_tile/host.hpp" #include "ck_tile/host.hpp"
#include "ck_tile/host/kernel_launch.hpp" #include "ck_tile/host/util/kernel_launch.hpp"
#include "ck_tile/ops/gemm.hpp" #include "ck_tile/device/ops/gemm.hpp"
#include "ck_tile/ops/epilogue.hpp" #include "ck_tile/device/ops/epilogue.hpp"
template <typename DataType> template <typename DataType>
struct GemmBasicTypeConfig; struct GemmBasicTypeConfig;
......
// 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/kernel_launch.hpp" #include "ck_tile/host/util/kernel_launch.hpp"
#include "ck_tile/ops/image_to_column.hpp" #include "ck_tile/device/ops/image_to_column.hpp"
#include <string> #include <string>
#define DefaultConvParams \ #define DefaultConvParams \
......
// 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/common.hpp" #include "ck_tile/device/ops/common.hpp"
#include "ck_tile/ops/reduce/block/block_reduce.hpp" #include "ck_tile/device/ops/reduce/block/block_reduce.hpp"
#include "ck_tile/ops/reduce/block/block_reduce2d_default_policy.hpp" #include "ck_tile/device/ops/reduce/block/block_reduce2d_default_policy.hpp"
namespace ck_tile { namespace ck_tile {
......
// 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.hpp" #include "ck_tile/host.hpp"
#include "ck_tile/ops/gemm.hpp" #include "ck_tile/device/ops/gemm.hpp"
// if set to 1, slightly more instructions generated to calculate address // if set to 1, slightly more instructions generated to calculate address
#ifndef MERGE_2D_013425 #ifndef MERGE_2D_013425
......
// 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/kernel_launch.hpp" #include "ck_tile/host/util/kernel_launch.hpp"
#include "ck_tile/ops/permute.hpp" #include "ck_tile/device/ops/permute.hpp"
#include <string> #include <string>
struct permute_traits struct permute_traits
......
// 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.
#include <vector> #include <vector>
#include <iostream> #include <iostream>
...@@ -11,7 +11,7 @@ ...@@ -11,7 +11,7 @@
#include <unordered_set> #include <unordered_set>
#include "ck_tile/core.hpp" #include "ck_tile/core.hpp"
#include "ck_tile/ops/reduce.hpp" #include "ck_tile/device/ops/reduce.hpp"
#include "topk_softmax_api.hpp" #include "topk_softmax_api.hpp"
#if 0 #if 0
......
// 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.hpp" #include "ck_tile/host.hpp"
#include "ck_tile/ops/topk_softmax.hpp" #include "ck_tile/device/ops/topk_softmax.hpp"
#include <string> #include <string>
struct topk_softmax_trait struct topk_softmax_trait
......
#include "ck_tile/host.hpp" #include "ck_tile/host.hpp"
#include "ck_tile/core.hpp" #include "ck_tile/core.hpp"
#include "ck_tile/host/kernel_launch.hpp" #include "ck_tile/host/util/kernel_launch.hpp"
#include "ck_tile/ops/rmsnorm2d.hpp" #include "ck_tile/device/ops/rmsnorm2d.hpp"
#include <cstring> #include <cstring>
auto create_args(int argc, char* argv[]) auto create_args(int argc, char* argv[])
......
// 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/kernel_launch.hpp" #include "ck_tile/host/util/kernel_launch.hpp"
#include "ck_tile/ops/rmsnorm2d.hpp" #include "ck_tile/device/ops/rmsnorm2d.hpp"
#include <string> #include <string>
template <typename DataType> template <typename DataType>
......
// 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/kernel_launch.hpp" #include "ck_tile/host/util/kernel_launch.hpp"
#include "ck_tile/ops/add_rmsnorm2d_rdquant.hpp" #include "ck_tile/device/ops/add_rmsnorm2d_rdquant.hpp"
#include <string> #include <string>
template <typename DataType> template <typename DataType>
......
#include "ck_tile/host.hpp" #include "ck_tile/host.hpp"
#include "ck_tile/core.hpp" #include "ck_tile/core.hpp"
#include "ck_tile/host/kernel_launch.hpp" #include "ck_tile/host/util/kernel_launch.hpp"
#include "ck_tile/ops/add_rmsnorm2d_rdquant.hpp" #include "ck_tile/device/ops/add_rmsnorm2d_rdquant.hpp"
#include <cstring> #include <cstring>
// different threshold for different dtype // different threshold for different dtype
......
#include "ck_tile/host.hpp" #include "ck_tile/host.hpp"
#include "ck_tile/core.hpp" #include "ck_tile/core.hpp"
#include "ck_tile/host/kernel_launch.hpp" #include "ck_tile/host/util/kernel_launch.hpp"
#include "ck_tile/ops/smoothquant.hpp" #include "ck_tile/device/ops/smoothquant.hpp"
#include <cstring> #include <cstring>
// different threshold for different dtype // different threshold for different dtype
......
// 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/kernel_launch.hpp" #include "ck_tile/host/util/kernel_launch.hpp"
#include "ck_tile/ops/smoothquant.hpp" #include "ck_tile/device/ops/smoothquant.hpp"
#include <string> #include <string>
template <typename DataType> template <typename DataType>
......
// 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.
#include <set> #include <set>
#include <vector> #include <vector>
...@@ -12,7 +12,7 @@ ...@@ -12,7 +12,7 @@
#include <unordered_set> #include <unordered_set>
#include "ck_tile/core.hpp" #include "ck_tile/core.hpp"
#include "ck_tile/ops/reduce.hpp" #include "ck_tile/device/ops/reduce.hpp"
#include "moe_sorting_api.hpp" #include "moe_sorting_api.hpp"
auto create_args(int argc, char* argv[]) auto create_args(int argc, char* argv[])
......
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