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
e53b50e8
Commit
e53b50e8
authored
Sep 06, 2022
by
Po-Yen, Chen
Browse files
Create 'DevicePermuteBase' to generate methods
parent
ea343345
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
12 additions
and
1 deletion
+12
-1
include/ck/tensor_operation/gpu/device/device_permute.hpp
include/ck/tensor_operation/gpu/device/device_permute.hpp
+12
-1
No files found.
include/ck/tensor_operation/gpu/device/device_permute.hpp
View file @
e53b50e8
...
@@ -18,6 +18,11 @@ namespace ck {
...
@@ -18,6 +18,11 @@ namespace ck {
namespace
tensor_operation
{
namespace
tensor_operation
{
namespace
device
{
namespace
device
{
template
<
typename
Derived
>
struct
DevicePermuteBase
:
BaseOperator
{
};
template
<
typename
InDataTypeTuple
,
template
<
typename
InDataTypeTuple
,
typename
OutDataTypeTuple
,
typename
OutDataTypeTuple
,
typename
ElementwiseOperation
,
typename
ElementwiseOperation
,
...
@@ -25,7 +30,13 @@ template <typename InDataTypeTuple,
...
@@ -25,7 +30,13 @@ template <typename InDataTypeTuple,
index_t
MPerThread
,
index_t
MPerThread
,
typename
InScalarPerVectorSeq
,
typename
InScalarPerVectorSeq
,
typename
OutScalarPerVectorSeq
>
typename
OutScalarPerVectorSeq
>
struct
DevicePermute
:
BaseOperator
struct
DevicePermute
:
DevicePermuteBase
<
DevicePermute
<
InDataTypeTuple
,
OutDataTypeTuple
,
ElementwiseOperation
,
NumDim
,
MPerThread
,
InScalarPerVectorSeq
,
OutScalarPerVectorSeq
>>
{
{
static
constexpr
int
NumInput
=
InDataTypeTuple
::
Size
();
static
constexpr
int
NumInput
=
InDataTypeTuple
::
Size
();
static
constexpr
int
NumOutput
=
OutDataTypeTuple
::
Size
();
static
constexpr
int
NumOutput
=
OutDataTypeTuple
::
Size
();
...
...
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