permute_fp16.cpp 850 Bytes
Newer Older
Po-Yen, Chen's avatar
Po-Yen, Chen committed
1
2
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-2022, Advanced Micro Devices, Inc. All rights reserved.
3

Po-Yen, Chen's avatar
Po-Yen, Chen committed
4
#include "common.hpp"
5
6
7
8

using ADataType = F16;
using BDataType = F16;

9
10
11
12
13
14
15
16
// clang-format off
using DevicePermuteInstance = ck::tensor_operation::device::DevicePermute
// ######|    InData|   OutData| Elementwise| NumDim| MPerThread|  InScalar| OutScalar|
// ######|      Type|      Type|   Operation|       |           | PerVector| PerVector|
// ######|          |          |            |       |           |          |          |
// ######|          |          |            |       |           |          |          |
         < ADataType, BDataType, PassThrough,      4,          8,         8,         1>;
// clang-format on
17

18
#include "run_permute_example.inc"
19

20
int main(int argc, char* argv[]) { return !run_permute_example(argc, argv); }