Commit 756251c7 authored by Po-Yen, Chen's avatar Po-Yen, Chen
Browse files

Add example for 4xfp16 permute

parent 3e605990
...@@ -2,8 +2,10 @@ add_custom_target(example_permute) ...@@ -2,8 +2,10 @@ add_custom_target(example_permute)
add_example_executable(example_permute_1xHxW_fp32 permute_1xHxW_fp32.cpp) add_example_executable(example_permute_1xHxW_fp32 permute_1xHxW_fp32.cpp)
add_example_executable(example_permute_NxHxW_fp32 permute_NxHxW_fp32.cpp) add_example_executable(example_permute_NxHxW_fp32 permute_NxHxW_fp32.cpp)
add_example_executable_no_testing(example_permute_HxWx4_fp16 permute_HxWx4_fp16.cpp)
add_example_executable(example_permute_HxWx2_fp16 permute_HxWx2_fp16.cpp) add_example_executable(example_permute_HxWx2_fp16 permute_HxWx2_fp16.cpp)
add_dependencies(example_permute example_permute_1xHxW_fp32) add_dependencies(example_permute example_permute_1xHxW_fp32)
add_dependencies(example_permute example_permute_NxHxW_fp32) add_dependencies(example_permute example_permute_NxHxW_fp32)
add_dependencies(example_permute example_permute_HxWx4_fp16)
add_dependencies(example_permute example_permute_HxWx2_fp16) add_dependencies(example_permute example_permute_HxWx2_fp16)
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-2022, Advanced Micro Devices, Inc. All rights reserved.
#include "common.hpp"
using ADataType = F64;
using BDataType = F64;
// clang-format off
using DevicePermuteInstance = ck::tensor_operation::device::DevicePermute
// ######| InData| OutData| Elementwise| NumDim| Block| HPer| WPer| InBlock| InBlockTransfer| InBlockTransfer| Src| Dst| Src| Dst|
// ######| Type| Type| Operation| | Size| Block| Block| LdsExtraW| ThreadClusterLengths| ThreadClusterArrangeOrder| VectorDim| VectorDim| ScalarPerVector| ScalarPerVector|
// ######| | | | | | | | | | | | | | |
// ######| | | | | | | | | | | | | | |
< ADataType, BDataType, PassThrough, 3, 256, 128, 128, 0, S<1, 16, 16>, S<0, 1, 2>, 2, 1, 1, 1>;
// clang-format on
#define NUM_ELEMS_IN_BUNDLE 4
#include "run_permute_example.inc"
int main(int argc, char* argv[]) { return !run_permute_example(argc, argv, {1, 6, 5}, {0, 2, 1}); }
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