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
f015e568
"docs/vscode:/vscode.git/clone" did not exist on "eda72e756ec6fa984673bcb8ef1f2614f347c472"
Commit
f015e568
authored
Sep 06, 2022
by
Po-Yen, Chen
Browse files
Remove base class of 'DevicePermute'
parent
6c4268f9
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
7 deletions
+3
-7
include/ck/tensor_operation/gpu/device/device_permute.hpp
include/ck/tensor_operation/gpu/device/device_permute.hpp
+3
-7
No files found.
include/ck/tensor_operation/gpu/device/device_permute.hpp
View file @
f015e568
...
@@ -26,7 +26,6 @@ template <typename InDataTypeTuple,
...
@@ -26,7 +26,6 @@ template <typename InDataTypeTuple,
typename
InScalarPerVectorSeq
,
typename
InScalarPerVectorSeq
,
typename
OutScalarPerVectorSeq
>
typename
OutScalarPerVectorSeq
>
struct
DevicePermute
struct
DevicePermute
:
public
DeviceElementwiseBase
<
InDataTypeTuple
,
OutDataTypeTuple
,
ElementwiseOperation
,
NumDim
>
{
{
static
constexpr
int
NumInput
=
InDataTypeTuple
::
Size
();
static
constexpr
int
NumInput
=
InDataTypeTuple
::
Size
();
static
constexpr
int
NumOutput
=
OutDataTypeTuple
::
Size
();
static
constexpr
int
NumOutput
=
OutDataTypeTuple
::
Size
();
...
@@ -255,7 +254,7 @@ struct DevicePermute
...
@@ -255,7 +254,7 @@ struct DevicePermute
return
valid
;
return
valid
;
};
};
bool
IsSupportedArgument
(
const
BaseArgument
*
p_arg
)
override
bool
IsSupportedArgument
(
const
BaseArgument
*
p_arg
)
{
{
return
IsSupportedArgument
(
*
dynamic_cast
<
const
Argument
*>
(
p_arg
));
return
IsSupportedArgument
(
*
dynamic_cast
<
const
Argument
*>
(
p_arg
));
}
}
...
@@ -282,7 +281,7 @@ struct DevicePermute
...
@@ -282,7 +281,7 @@ struct DevicePermute
const
std
::
array
<
std
::
array
<
index_t
,
NumDim
>
,
NumOutput
>
outStridesArray
,
const
std
::
array
<
std
::
array
<
index_t
,
NumDim
>
,
NumOutput
>
outStridesArray
,
const
std
::
array
<
const
void
*
,
NumInput
>
in_dev_buffers
,
const
std
::
array
<
const
void
*
,
NumInput
>
in_dev_buffers
,
const
std
::
array
<
void
*
,
NumOutput
>
out_dev_buffers
,
const
std
::
array
<
void
*
,
NumOutput
>
out_dev_buffers
,
ElementwiseOperation
elementwise_op
)
override
ElementwiseOperation
elementwise_op
)
{
{
return
std
::
make_unique
<
Argument
>
(
lengths
,
return
std
::
make_unique
<
Argument
>
(
lengths
,
inStridesArray
,
inStridesArray
,
...
@@ -293,10 +292,7 @@ struct DevicePermute
...
@@ -293,10 +292,7 @@ struct DevicePermute
}
}
static
auto
MakeInvoker
()
{
return
Invoker
{};
}
static
auto
MakeInvoker
()
{
return
Invoker
{};
}
std
::
unique_ptr
<
BaseInvoker
>
MakeInvokerPointer
()
override
std
::
unique_ptr
<
BaseInvoker
>
MakeInvokerPointer
()
{
return
std
::
make_unique
<
Invoker
>
();
};
{
return
std
::
make_unique
<
Invoker
>
();
};
};
// namespace device
};
// namespace device
}
// namespace device
}
// namespace device
...
...
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