permute.hpp 459 Bytes
Newer Older
1
// SPDX-License-Identifier: MIT
2
// Copyright (c) 2018-2025, Advanced Micro Devices, Inc. All rights reserved.
3
4
5
6

#pragma once

#include "ck_tile/core.hpp"
7
8
#include "ck_tile/host/util/kernel_launch.hpp"
#include "ck_tile/device/ops/permute.hpp"
9
10
11
12
13
14
15
16
17
18
19
#include <string>

struct permute_traits
{
    std::string data_type;
};

using permute_args = ck_tile::GenericPermuteHostArgs;

// host API
float permute(permute_traits, permute_args, const ck_tile::stream_config&);