Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
gaoqiong
composable_kernel
Commits
756251c7
Commit
756251c7
authored
Sep 12, 2022
by
Po-Yen, Chen
Browse files
Add example for 4xfp16 permute
parent
3e605990
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
23 additions
and
0 deletions
+23
-0
example/36_permute/CMakeLists.txt
example/36_permute/CMakeLists.txt
+2
-0
example/36_permute/permute_HxWx4_fp16.cpp
example/36_permute/permute_HxWx4_fp16.cpp
+21
-0
No files found.
example/36_permute/CMakeLists.txt
View file @
756251c7
...
@@ -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
)
example/36_permute/permute_HxWx4_fp16.cpp
0 → 100644
View file @
756251c7
// 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
});
}
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment