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

[CK_TILE] Update the file structure part2

parent ec1b65e7
...@@ -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 \
......
...@@ -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
......
...@@ -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
......
#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[])
......
...@@ -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>
......
...@@ -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>
......
#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
......
#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
......
...@@ -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>
......
...@@ -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
{ {
......
...@@ -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>
......
...@@ -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
......
...@@ -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
{ {
......
...@@ -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>
......
...@@ -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 {
......
...@@ -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>
......
...@@ -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>
......
...@@ -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 {
......
...@@ -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 {
......
...@@ -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 {
......
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