mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-04 13:41:24 +00:00
21 lines
1.4 KiB
C++
21 lines
1.4 KiB
C++
// SPDX-License-Identifier: MIT
|
|
// Copyright (c) 2018-2023, Advanced Micro Devices, Inc. All rights reserved.
|
|
|
|
#include "common.hpp"
|
|
|
|
using InDataType = F16;
|
|
using OutDataType = F16;
|
|
|
|
// clang-format off
|
|
using DevicePermuteInstance = ck::tensor_operation::device::DevicePermuteImpl
|
|
// ######| NumDim| InData| OutData| Elementwise| Block| NPer| HPer| WPer| InBlock| InBlockTransfer| InBlockTransfer| Src| Dst| Src| Dst|
|
|
// ######| | Type| Type| Operation| Size| Block| Block| Block| LdsExtraW| ThreadClusterLengths| ThreadClusterArrangeOrder| VectorDim| VectorDim| ScalarPerVector| ScalarPerVector|
|
|
// ######| | | | | | | | | | | | | | | |
|
|
// ######| | | | | | | | | | | | | | | |
|
|
< 3, InDataType, OutDataType, PassThrough, 128, 4, 16, 8, 6, S<2, 16, 4>, S<0, 1, 2>, 2, 1, 2, 1>;
|
|
// clang-format on
|
|
|
|
#include "run_permute_element_example.inc"
|
|
|
|
int main() { return !run_permute_element_example({121, 768, 80}, {0, 2, 1}); }
|