Skip to content

GitLab

Commit bc26a2fa authored by Po-Yen, Chen's avatar Po-Yen, Chen
Browse files

Let 'DevicePermute' inherit from 'BaseOperator'

parent f015e568
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
include/ck/tensor_operation/gpu/device/device_permute.hpp
View file @ bc26a2fa
...@@ -25,7 +25,7 @@ template <typename InDataTypeTuple, ...@@ -25,7 +25,7 @@ template <typename InDataTypeTuple,
index_t MPerThread, index_t MPerThread,
typename InScalarPerVectorSeq, typename InScalarPerVectorSeq,
typename OutScalarPerVectorSeq> typename OutScalarPerVectorSeq>
struct DevicePermute struct DevicePermute : BaseOperator
{ {
static constexpr int NumInput = InDataTypeTuple::Size(); static constexpr int NumInput = InDataTypeTuple::Size();
static constexpr int NumOutput = OutDataTypeTuple::Size(); static constexpr int NumOutput = OutDataTypeTuple::Size();
...@@ -254,7 +254,7 @@ struct DevicePermute ...@@ -254,7 +254,7 @@ struct DevicePermute
return valid; return valid;
}; };
bool IsSupportedArgument(const BaseArgument* p_arg) bool IsSupportedArgument(const BaseArgument* p_arg) override
{ {
return IsSupportedArgument(*dynamic_cast<const Argument*>(p_arg)); return IsSupportedArgument(*dynamic_cast<const Argument*>(p_arg));
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment