mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-03 13:11:25 +00:00
20 lines
447 B
C++
20 lines
447 B
C++
// SPDX-License-Identifier: MIT
|
|
// Copyright (c) 2018-2024, Advanced Micro Devices, Inc. All rights reserved.
|
|
|
|
#pragma once
|
|
|
|
#include "ck_tile/core.hpp"
|
|
#include "ck_tile/host/kernel_launch.hpp"
|
|
#include "ck_tile/ops/permute.hpp"
|
|
#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&);
|