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
4eaa502b
Commit
4eaa502b
authored
Sep 08, 2022
by
Po-Yen, Chen
Browse files
Rename local type alias
parent
e5e7adbd
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
20 additions
and
20 deletions
+20
-20
include/ck/tensor_operation/gpu/grid/gridwise_permute.hpp
include/ck/tensor_operation/gpu/grid/gridwise_permute.hpp
+20
-20
No files found.
include/ck/tensor_operation/gpu/grid/gridwise_permute.hpp
View file @
4eaa502b
...
...
@@ -204,13 +204,13 @@ struct GridwisePermute
auto
in_block_buf
=
make_dynamic_buffer
<
AddressSpaceEnum
::
Lds
>
(
static_cast
<
InDataType
*>
(
p_shared
),
in_block_desc
.
GetElementSpaceSize
());
using
SliceLengths
=
Sequence
<
1
,
HPerBlock
,
WPerBlock
>
;
using
A
BlockTransferAccessOrder
=
Sequence
<
0
,
1
,
2
>
;
using
Block
SliceLengths
=
Sequence
<
1
,
HPerBlock
,
WPerBlock
>
;
using
In
BlockTransferAccessOrder
=
Sequence
<
0
,
1
,
2
>
;
constexpr
index_t
ABlockTransfer
SrcVectorDim
=
2
;
constexpr
index_t
ABlockTransfer
DstVectorDim
=
1
;
constexpr
index_t
ABlockTransfer
SrcScalarPerVector
=
1
;
constexpr
index_t
ABlockTransfer
DstScalarPerVector
=
1
;
constexpr
index_t
SrcVectorDim
=
2
;
constexpr
index_t
DstVectorDim
=
1
;
constexpr
index_t
SrcScalarPerVector
=
1
;
constexpr
index_t
DstScalarPerVector
=
1
;
using
ck
::
tensor_operation
::
element_wise
::
PassThrough
;
...
...
@@ -221,19 +221,19 @@ struct GridwisePermute
ElementwiseOperation
,
PassThrough
,
InMemoryDataOperationEnum
::
Set
,
SliceLengths
,
Block
SliceLengths
,
InBlockTransferThreadClusterLengths
,
InBlockTransferThreadClusterArrangeOrder
,
InDataType
,
InDataType
,
decltype
(
in_grid_desc_n_h_w
),
decltype
(
in_block_desc
),
A
BlockTransferAccessOrder
,
A
BlockTransferAccessOrder
,
ABlockTransfer
SrcVectorDim
,
ABlockTransfer
SrcVectorDim
,
ABlockTransfer
SrcScalarPerVector
,
ABlockTransfer
SrcScalarPerVector
,
In
BlockTransferAccessOrder
,
In
BlockTransferAccessOrder
,
SrcVectorDim
,
SrcVectorDim
,
SrcScalarPerVector
,
SrcScalarPerVector
,
1
,
1
,
true
,
...
...
@@ -260,19 +260,19 @@ struct GridwisePermute
ElementwiseOperation
,
PassThrough
,
InMemoryDataOperationEnum
::
Set
,
SliceLengths
,
Block
SliceLengths
,
InBlockTransferThreadClusterLengths
,
InBlockTransferThreadClusterArrangeOrder
,
InDataType
,
OutDataType
,
decltype
(
in_block_desc
),
decltype
(
out_grid_desc_n_h_w
),
A
BlockTransferAccessOrder
,
A
BlockTransferAccessOrder
,
ABlockTransfer
SrcVectorDim
,
ABlockTransfer
DstVectorDim
,
ABlockTransfer
SrcScalarPerVector
,
ABlockTransfer
DstScalarPerVector
,
In
BlockTransferAccessOrder
,
In
BlockTransferAccessOrder
,
SrcVectorDim
,
DstVectorDim
,
SrcScalarPerVector
,
DstScalarPerVector
,
1
,
1
,
true
,
...
...
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