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 @@
#include "ck_tile/core.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>
#define DefaultConvParams \
......
......@@ -5,7 +5,7 @@
#include "ck_tile/core.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>
struct permute_traits
......
......@@ -4,7 +4,7 @@
#pragma once
#include "ck_tile/core.hpp"
#include "ck_tile/host.hpp"
#include "ck_tile/device/ops/topk_softmax.hpp"
#include "ck_tile/host/ops/topk_softmax.hpp"
#include <string>
struct topk_softmax_trait
......
#include "ck_tile/host.hpp"
#include "ck_tile/core.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>
auto create_args(int argc, char* argv[])
......
......@@ -5,7 +5,7 @@
#include "ck_tile/core.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>
template <typename DataType>
......
......@@ -5,7 +5,7 @@
#include "ck_tile/core.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>
template <typename DataType>
......
#include "ck_tile/host.hpp"
#include "ck_tile/core.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>
// different threshold for different dtype
......
#include "ck_tile/host.hpp"
#include "ck_tile/core.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>
// different threshold for different dtype
......
......@@ -5,7 +5,7 @@
#include "ck_tile/core.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>
template <typename DataType>
......
......@@ -5,7 +5,7 @@
#include <string>
#include "ck_tile/core.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
{
......
......@@ -5,7 +5,7 @@
#include "ck_tile/core.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>
template <typename DataType>
......
......@@ -5,7 +5,7 @@
#include "ck_tile/core.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>
// this is only a convenient structure for creating an example
......
......@@ -5,7 +5,7 @@
#include <string>
#include "ck_tile/core.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
{
......
......@@ -4,7 +4,7 @@
#pragma once
#include "fused_moegemm_api_traits.hpp"
#include "ck_tile/device/ops/fused_moe.hpp"
#include "ck_tile/host/ops/fused_moe.hpp"
#include <iostream>
template <ck_tile::index_t... Is>
......
......@@ -4,8 +4,8 @@
#pragma once
#include "ck_tile/core.hpp"
#include "ck_tile/device/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_problem.hpp"
#include "ck_tile/host/ops/norm_reduce/block/block_norm_reduce.hpp"
namespace ck_tile {
......
......@@ -4,7 +4,7 @@
#pragma once
#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 <type_traits>
......
......@@ -4,7 +4,7 @@
#pragma once
#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 <type_traits>
......
......@@ -5,7 +5,7 @@
#include "ck_tile/core.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 {
......
......@@ -5,7 +5,7 @@
#include "ck_tile/core.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 {
......
......@@ -5,8 +5,8 @@
#include "ck_tile/core.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/device/ops/flatmm/block/flatmm_sn_32x128x512_1x4x1_16x16x32.hpp"
#include "ck_tile/host/ops/flatmm/block/flatmm_uk_config.hpp"
#include "ck_tile/host/ops/flatmm/block/flatmm_sn_32x128x512_1x4x1_16x16x32.hpp"
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