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
e8b0e222
Commit
e8b0e222
authored
Sep 12, 2022
by
Po-Yen, Chen
Browse files
Avoid using too-large LDS in kernel code
parent
0a2131e9
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
7 additions
and
4 deletions
+7
-4
example/37_permute/CMakeLists.txt
example/37_permute/CMakeLists.txt
+2
-2
example/37_permute/permute_HxWx4_fp16.cpp
example/37_permute/permute_HxWx4_fp16.cpp
+5
-2
No files found.
example/37_permute/CMakeLists.txt
View file @
e8b0e222
...
...
@@ -2,10 +2,10 @@ add_custom_target(example_permute)
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
_no_testing
(example_permute_HxWx4_fp16 permute_HxWx4_fp16.cpp)
add_example_executable
(
example_permute_HxWx4_fp16 permute_HxWx4_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_NxHxW_fp32
)
#
add_dependencies(example_permute example_permute_HxWx4_fp16)
add_dependencies
(
example_permute example_permute_HxWx4_fp16
)
add_dependencies
(
example_permute example_permute_HxWx2_fp16
)
example/37_permute/permute_HxWx4_fp16.cpp
View file @
e8b0e222
...
...
@@ -12,10 +12,13 @@ using DevicePermuteInstance = ck::tensor_operation::device::DevicePermute
// ######| 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
>
;
<
ADataType
,
BDataType
,
PassThrough
,
3
,
256
,
16
,
16
,
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
});
}
int
main
(
int
argc
,
char
*
argv
[])
{
return
!
run_permute_example
(
argc
,
argv
,
{
1
,
64
,
32
},
{
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